Update README.md
Browse files
README.md
CHANGED
|
@@ -28,6 +28,7 @@ This is a version of the <a href="https://huggingface.co/deepseek-ai/DeepSeek-R1
|
|
| 28 |
|
| 29 |
## Usage
|
| 30 |
```Python
|
|
|
|
| 31 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 32 |
compute_dtype = torch.bfloat16
|
| 33 |
device = 'cuda'
|
|
|
|
| 28 |
|
| 29 |
## Usage
|
| 30 |
```Python
|
| 31 |
+
import torch
|
| 32 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 33 |
compute_dtype = torch.bfloat16
|
| 34 |
device = 'cuda'
|