Views
No views yet
• Model Type: Causal Language Model (LLM)
• Base Model: Meta-Llama-3.1-8B
• Architecture: Transformer-based autoregressive model
• Quantization: 4-bit precision using BitsAndBytes for memory efficiency
• Training Method: LoRA fine-tuning
• Task: General language generation, conversation, text completion• Conversational AI assistants
• Text completion
• Response generation in chatbots
• Any task that involves understanding and generating human-like text• LoRA Configuration:
• r=8, lora_alpha=16, lora_dropout=0.05
• This setup introduces efficient low-rank adaptation to improve model training with a smaller number of parameters.
• Training Arguments:
• Batch size per device: 4
• Learning rate: 2e-4
• Training epochs: 3
• Gradient accumulation: 16 steps
• Optimizer: paged_adamw_32bit
• Fine-tuning on custom dataset using Trainer with push to Hugging Face Hub.
• Quantization:
• Load in 4-bit precision (bnb_4bit), quantization type: nf4
• The model is optimized for efficient inference using float16 compute precision.• Generation Config:
• penalty_alpha=0.6
• do_sample=True
• top_k=5
• temperature=0.5
• repetition_penalty=1.2
• max_new_tokens=60• Hardware Requirements:
• 8-bit/4-bit quantization allows the model to run on consumer-grade GPUs with efficient memory utilization.
• Inference Time:
• Response generation time varies based on prompt complexity but typically completes within 2-4 seconds on a standard GPU setup.• The model might generate biased or inappropriate content as it is trained on publicly available datasets and could reflect biases inherent in those datasets.
• Proper filtering and human supervision are recommended for sensitive use cases, such as those involving ethical or safety-critical scenarios.