IlyaGusev
/

saiga_mistral_7b_lora

Text Generation

Model card Files Files and versions Community

IlyaGusev commited on Oct 9, 2023

Commit

7403592

•

1 Parent(s): ae7552f

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -24,6 +24,7 @@ Colab: TBA
 Training code: [link](https://github.com/IlyaGusev/rulm/tree/master/self_instruct).
 ```python
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig

 Training code: [link](https://github.com/IlyaGusev/rulm/tree/master/self_instruct).
 ```python
+import torch
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig