vibhorag101
/

llama-2-7b-chat-hf-phr_mental_therapy

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

vibhorag101 commited on Nov 9, 2023

Commit

e9b8994

•

1 Parent(s): bbf79b9

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -20,7 +20,7 @@ pipeline_tag: text-generation
 - 48 Core Intel Xeon
 - 128GB RAM.
 ### Model Hyperparameters
-- This [training script](https://github.com/vibhorag101/phr-chat/blob/main/Finetune/finetune.ipynb) was used to do the finetuning.
 - The shareGPT format dataset was converted to llama-2 training format using this [script](https://github.com/vibhorag101/phr-chat/blob/main/Finetune/data_transform.ipynb).
 - num_train_epochs = 3
 - per_device_train_batch_size = 2

 - 48 Core Intel Xeon
 - 128GB RAM.
 ### Model Hyperparameters
+- This [training script](https://github.com/vibhorag101/llama2-mental-therapy/blob/main/finetuneModel/finetuneScript.ipynb) was used to do the finetuning.
 - The shareGPT format dataset was converted to llama-2 training format using this [script](https://github.com/vibhorag101/phr-chat/blob/main/Finetune/data_transform.ipynb).
 - num_train_epochs = 3
 - per_device_train_batch_size = 2