update readme and add typehints

This commit is contained in:
Wing Lian
2023-05-25 23:10:11 -04:00
parent 48f4c0571e
commit a4f12415a0
2 changed files with 9 additions and 14 deletions

View File

@@ -363,13 +363,7 @@ Pass the appropriate flag to the train command:
### Merge LORA to base
Add below flag to train command above (and using LoRA)
```bash
--merge_lora --lora_model_dir="./completed-model"
```
Add below flag to train command above (and using QLoRA)
Add below flag to train command above
```bash
--merge_lora --lora_model_dir="./completed-model" --load_in_8bit=False --load_in_4bit=False