Update README.md
Browse files
README.md
CHANGED
|
@@ -53,7 +53,7 @@ Make sure to update your transformers installation via `pip install --upgrade tr
|
|
| 53 |
import transformers
|
| 54 |
import torch
|
| 55 |
|
| 56 |
-
model_id = "MichiganNLP/
|
| 57 |
|
| 58 |
pipeline = transformers.pipeline(
|
| 59 |
"text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto"
|
|
|
|
| 53 |
import transformers
|
| 54 |
import torch
|
| 55 |
|
| 56 |
+
model_id = "MichiganNLP/TAMA-vA"
|
| 57 |
|
| 58 |
pipeline = transformers.pipeline(
|
| 59 |
"text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto"
|