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 = "Напиши стих про машинное обучение"
|