Update README.md
#7
by fliiink - opened
README.md
CHANGED
|
@@ -25,6 +25,7 @@ llama.cpp version: [link](https://huggingface.co/IlyaGusev/saiga_7b_ggml)
|
|
| 25 |
Training code: [link](https://github.com/IlyaGusev/rulm/tree/master/self_instruct)
|
| 26 |
|
| 27 |
```python
|
|
|
|
| 28 |
from peft import PeftModel, PeftConfig
|
| 29 |
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
|
| 30 |
|
|
|
|
| 25 |
Training code: [link](https://github.com/IlyaGusev/rulm/tree/master/self_instruct)
|
| 26 |
|
| 27 |
```python
|
| 28 |
+
import torch
|
| 29 |
from peft import PeftModel, PeftConfig
|
| 30 |
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
|
| 31 |
|