Update README.md
Browse files
README.md
CHANGED
|
@@ -9,6 +9,7 @@ This repo contains a low-rank adapter for LLaMA-13b fit on the Stanford Alpaca d
|
|
| 9 |
### How to use (8-bit)
|
| 10 |
|
| 11 |
```python
|
|
|
|
| 12 |
from peft import PeftModel
|
| 13 |
from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
|
| 14 |
|
|
|
|
| 9 |
### How to use (8-bit)
|
| 10 |
|
| 11 |
```python
|
| 12 |
+
import torch
|
| 13 |
from peft import PeftModel
|
| 14 |
from transformers import LlamaTokenizer, LlamaForCausalLM, GenerationConfig
|
| 15 |
|