vibhorag101 commited on
Commit
42bd259
1 Parent(s): e9b8994

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -21,7 +21,7 @@ pipeline_tag: text-generation
21
  - 128GB RAM.
22
  ### Model Hyperparameters
23
  - This [training script](https://github.com/vibhorag101/llama2-mental-therapy/blob/main/finetuneModel/finetuneScript.ipynb) was used to do the finetuning.
24
- - 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).
25
  - num_train_epochs = 3
26
  - per_device_train_batch_size = 2
27
  - per_device_eval_batch_size = 2
 
21
  - 128GB RAM.
22
  ### Model Hyperparameters
23
  - This [training script](https://github.com/vibhorag101/llama2-mental-therapy/blob/main/finetuneModel/finetuneScript.ipynb) was used to do the finetuning.
24
+ - The shareGPT format dataset was converted to llama-2 training format using this [script](https://github.com/vibhorag101/llama2-mental-therapy/blob/main/finetuneModel/llamaDataMaker.ipynb).
25
  - num_train_epochs = 3
26
  - per_device_train_batch_size = 2
27
  - per_device_eval_batch_size = 2