germanjke commited on
Commit
aa2b15d
1 Parent(s): 718ca73

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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-ai/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 = "Напиши стих про машинное обучение"
 
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 = "Напиши стих про машинное обучение"