germanjke commited on
Commit
3bdc95b
1 Parent(s): aa2b15d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -14,8 +14,8 @@ 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 = "Напиши стих про машинное обучение"
20
  messages = [
21
  {"role": "system", "content": "Ты T-pro, виртуальный ассистент в Т-Технологии. Твоя задача - быть полезным диалоговым ассистентом."},
 
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)
18
+ model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
19
  prompt = "Напиши стих про машинное обучение"
20
  messages = [
21
  {"role": "system", "content": "Ты T-pro, виртуальный ассистент в Т-Технологии. Твоя задача - быть полезным диалоговым ассистентом."},