Update README.md
Browse files
README.md
CHANGED
|
@@ -13,7 +13,7 @@ Detailed model card’s coming soon…
|
|
| 13 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 14 |
import torch
|
| 15 |
torch.manual_seed(42)
|
| 16 |
-
model_name = "t-tech
|
| 17 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=token)
|
| 18 |
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=token, device_map="auto")
|
| 19 |
prompt = "Напиши стих про машинное обучение"
|
|
|
|
| 13 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 14 |
import torch
|
| 15 |
torch.manual_seed(42)
|
| 16 |
+
model_name = "t-tech/T-pro-it-1.0"
|
| 17 |
tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=token)
|
| 18 |
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=token, device_map="auto")
|
| 19 |
prompt = "Напиши стих про машинное обучение"
|