Update README.md
Browse files
README.md
CHANGED
|
@@ -18,6 +18,9 @@ This is the model card of a 🤗 transformers model that has been pushed on the
|
|
| 18 |
### Direct Use
|
| 19 |
|
| 20 |
```python
|
|
|
|
|
|
|
|
|
|
| 21 |
model = AutoModelForCausalLM.from_pretrained('FrankL/storytellerLM-v0.1', trust_remote_code=True, torch_dtype=torch.float16)
|
| 22 |
model = model.to(device='cuda')
|
| 23 |
|
|
|
|
| 18 |
### Direct Use
|
| 19 |
|
| 20 |
```python
|
| 21 |
+
import torch
|
| 22 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 23 |
+
|
| 24 |
model = AutoModelForCausalLM.from_pretrained('FrankL/storytellerLM-v0.1', trust_remote_code=True, torch_dtype=torch.float16)
|
| 25 |
model = model.to(device='cuda')
|
| 26 |
|