student-abdullah's picture
Update README.md
2d64852 verified
|
raw
history blame
2.04 kB
metadata
base_model: meta-llama/Llama-3.2-1B
datasets:
  - student-abdullah/BigPharma_Generic_Q-A_Format_Augemented_Dataset
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: 8th 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: 1024
  • LoRA Alpha: 4
  • LoRA Rank (r): 256
  • Learning rate: 5e-5
  • Gradient Accumulation Steps: 1
  • Batch Size: 8
  • Quantization: None

Model Quantitative Performace

  • Training Quantitative Loss: 0.1375 (at final 10rd epoch 9020nd 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: 83.20%
  • Incorrect Responses: 16.80%