Downloads · 30 days
0
Mubin1917/lamini_docs_LoRA_adapter
lamini_docs_LoRA_adapter is a machine learning model from Mubin1917. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
- Developed by: Mubin1917 - License: apache-2.0 - Finetuned from model: unsloth/meta-llama-3.1-8b-instruct-bnb-4bit
Downloads · 30 days
0
Access
Public
Updated Aug 21, 2024
Repo size
168 MB
Likes
0
Public
Click a slice to open those files.
.safetensors168 MB · 100%
From the Hugging Face model README
This LORA adapter is based on the unsloth/meta-llama-3.1-8b-instruct-bnb-4bit model and has been fine-tuned on the Lamini_docs QnA dataset. The fine-tuning process was optimized using Unsloth and Huggingface's TRL library, resulting in a 2x faster training time.
The model was trained with the following configuration:
training_args = TrainingArguments(
num_train_epochs=6,
per_device_train_batch_size=4,
gradient_accumulation_steps=4,
per_device_eval_batch_size=4,
eval_accumulation_steps=4,
warmup_steps=50,
learning_rate=2e-4,
fp16=not torch.cuda.is_bf16_supported(),
bf16=torch.cuda.is_bf16_supported(),
eval_steps=25, # Evaluate every 25 steps
logging_steps=25,
optim="adamw_8bit",
weight_decay=0.01,
lr_scheduler_type="linear",
seed=3407,
output_dir="/kaggle/temp/results",
report_to="wandb",
save_total_limit=1, # Save the best one and the last one
metric_for_best_model="val_loss",
eval_strategy="steps",
load_best_model_at_end=True,
)
SacreBLEU Test:
Score: 73.55
Detailed Metrics:
BLEU Test:
BLEU Score: 0.767
Detailed Metrics:
For a detailed comparison between the predicted and actual QnA responses on the test dataset, please visit the evaluation dataset.