Commit
•
a91a5d1
1
Parent(s):
0c0435c
Update README.md
Browse files
README.md
CHANGED
|
@@ -23,6 +23,7 @@ It is fine-tuned on the [VidProM](https://huggingface.co/datasets/WenhaoWang/Vid
|
|
| 23 |
## Download the model
|
| 24 |
```
|
| 25 |
from transformers import pipeline
|
|
|
|
| 26 |
pipe = pipeline("text-generation", model="WenhaoWang/Meta-Llama-3-8B-AutoT2VPrompt", model_kwargs={"torch_dtype": torch.bfloat16}, device_map="cuda:0")
|
| 27 |
```
|
| 28 |
|
|
|
|
| 23 |
## Download the model
|
| 24 |
```
|
| 25 |
from transformers import pipeline
|
| 26 |
+
import torch
|
| 27 |
pipe = pipeline("text-generation", model="WenhaoWang/Meta-Llama-3-8B-AutoT2VPrompt", model_kwargs={"torch_dtype": torch.bfloat16}, device_map="cuda:0")
|
| 28 |
```
|
| 29 |
|