File size: 1,020 Bytes
07f0fc0 998a3c0 07f0fc0 a2567d0 4c4c95e 7f2164c fd3d33b 7f2164c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
---
tags:
- autotrain
- text-generation
widget:
- text: 'What are the visiting hours for Adiyogi? '
datasets:
- nateshmbhat/isha-qa-text
pipeline_tag: text-generation
---
# ISHA Call Center QA Model
## This model was trained on a finetuned version(StableBeluga2) of Llama2-13B from stabilityai : [(StableBeluga2 tops the LLM leaderboard currently)](https://huggingface.co/stabilityai/StableBeluga2)
### Dataset Used : https://huggingface.co/datasets/nateshmbhat/isha-qa-text
#### Train Params used :
- Base model : stabilityai/StableBeluga-13B
- Quantization Used : 4 bit
- Learning rate : 2e-4
- Batch Size : 2
- Epochs : 3
- Trainer : sft
- Max token length : 2048 (capable of higher token length)
####
```
!autotrain llm --train --project_name project-isha-qa --model stabilityai/StableBeluga-13B --data_path nateshmbhat/isha-qa-text --use_peft --use_int4 --learning_rate 2e-4 --train_batch_size 2 --num_train_epochs 3 --trainer sft --model_max_length 2048 --push_to_hub --repo_id nateshmbhat/model-isha-qa
``` |