Views
No views yet
<|user|>, <|assistant|>).<|user|>
вопрос
<|assistant|>
ответtrust_remote_code=True.1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4repo_id = "GoldenGekko/LinguaLaboratoriumMechanicus-instruct"
5device = "cuda" if torch.cuda.is_available() else "cpu"
6
7question = "Что такое Гибельный шторм и как он повлиял на ход войны?"
8prompt = f"<|user|>\n{question}\n<|assistant|>\n"
9
10tokenizer = AutoTokenizer.from_pretrained(repo_id)
11model = AutoModelForCausalLM.from_pretrained(repo_id, trust_remote_code=True).to(device)
12
13inputs = tokenizer(prompt, return_tensors="pt").to(device)
14output_ids = model.generate(
15 inputs["input_ids"],
16 max_new_tokens=120,
17 do_sample=False,
18 pad_token_id=tokenizer.pad_token_id,
19 eos_token_id=tokenizer.eos_token_id,
20)
21print(tokenizer.decode(output_ids[0], skip_special_tokens=True).split("<|assistant|>")[-1].strip())https://huggingface.co/GoldenGekko/LinguaLaboratoriumMechanicus-instruct