t-tech
/

T-lite-it-1.0

Model card Files Files and versions Community

germanjke commited on 12 days ago

Commit

93b6151

•

1 Parent(s): a186d65

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -15,8 +15,8 @@ import torch
 torch.manual_seed(42)
 model_name = "t-tech/T-lite-it-1.0"
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=token)
-model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=token, device_map="auto")
 prompt = "Напиши стих про машинное обучение"
 messages = [

 torch.manual_seed(42)
 model_name = "t-tech/T-lite-it-1.0"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
 prompt = "Напиши стих про машинное обучение"
 messages = [