Update README.md
Browse files
README.md
CHANGED
|
@@ -52,6 +52,7 @@ pip install auto-gptq
|
|
| 52 |
```
|
| 53 |
|
| 54 |
```
|
|
|
|
| 55 |
from transformers import AutoTokenizer
|
| 56 |
from auto_gptq import AutoGPTQForCausalLM
|
| 57 |
|
|
|
|
| 52 |
```
|
| 53 |
|
| 54 |
```
|
| 55 |
+
import torch
|
| 56 |
from transformers import AutoTokenizer
|
| 57 |
from auto_gptq import AutoGPTQForCausalLM
|
| 58 |
|