student-abdullah's picture
Update README.md
c5be7df verified
|
raw
history blame
2 kB
metadata
base_model: meta-llama/Llama-3.2-1B
datasets:
  - student-abdullah/Experimental
language:
  - en
license: apache-2.0
tags:
  - text-generation-inference
  - transformers
  - torch
  - trl
  - unsloth
  - llama
  - gguf

Uploaded model

  • Developed by: student-abdullah
  • License: apache-2.0
  • Finetuned from model: meta-llama/Llama-3.2-1B
  • Created on: 3st October, 2024

Acknowledgement


Model Description

This model is fine-tuned from the meta-llama/Llama-3.2-1B base model to enhance its capabilities in generating relevant and accurate responses related to generic medications under the PMBJP scheme. The fine-tuning process included the following hyperparameters:

  • Fine Tuning Template: Llama Q&A
  • Max Tokens: 512
  • LoRA Alpha: 6
  • LoRA Rank (r): 128
  • Learning rate: 5e-5
  • Gradient Accumulation Steps: 2
  • Batch Size: 4
  • Quantization: 16 bits

Model Quantitative Performace

  • Training Quantitative Loss: 0.1407 (at final 5th epoch 5150th Step)

Limitations

  • Token Limitations: With a max token limit of 512, the model might not handle very long queries or contexts effectively.
  • Training Data Limitations: The model’s performance is contingent on the quality and coverage of the fine-tuning dataset, which may affect its generalizability to different contexts or medications not covered in the dataset.
  • Potential Biases: As with any model fine-tuned on specific data, there may be biases based on the dataset used for training.

Model Performace Evaluation:

  • Evaluation on 1000 Questions based on dataset (to evaluate the finetuned knowledge base)
  • At temperature 0.3
  • Correct Responses: 76%
  • Incorrect Responses: 24%