|
from transformers import MT5ForConditionalGeneration, MT5Tokenizer |
|
|
|
|
|
model_directory = './' |
|
|
|
|
|
tokenizer = MT5Tokenizer.from_pretrained(pretrained_model_name_or_path=model_directory, sp_model_kwargs={"model_file": f"{model_directory}/spiece.model"}) |
|
|
|
|
|
model = MT5ForConditionalGeneration.from_pretrained(pretrained_model_name_or_path=model_directory, local_files_only=True) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
context = "Вы бот PocketBook Reader. Вы говорите только о PocketBook Reader, и вам не разрешается сравнивать PocketBook с другими брендами, такими как Kindle, Tolino, Kobo и т. д." |
|
question = "Как включить PocketBook?" |
|
|
|
|
|
|
|
input_text = f"question: {question} context: {context}" |
|
input_ids = tokenizer(input_text, return_tensors="pt").input_ids |
|
|
|
|
|
outputs = model.generate(input_ids) |
|
answer = tokenizer.decode(outputs[0], skip_special_tokens=True) |
|
|
|
print(answer) |