--- 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 ```