Views
No views yet
1"""### Instruction: {instruction}
2
3### Response:
4"""1import os
2import torch
3import sys
4from transformers import LlamaForCausalLM, AutoTokenizer
5
6
7def generate_prompt_test(instruction):
8 prompt = f"""### Instruction: {instruction}
9
10### Response:
11"""
12 return prompt
13
14model_name = "raicrits/Hermes7b_ITA"
15
16model = LlamaForCausalLM.from_pretrained(
17 model_name,
18 device_map="auto",
19 torch_dtype=torch.bfloat16
20)
21
22model.config.use_cache = True
23
24
25tokenizer = AutoTokenizer.from_pretrained(model_name, add_eos_token=False)
26
27prompt = generate_prompt_test("Cosa puoi dirmi sul dio Hermes?")
28inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
29
30outputs = model.generate(**inputs, do_sample = True, num_beams = 2, top_k=50, top_p= 0.95, max_new_tokens=256, early_stopping = True)
31print(tokenizer.decode(outputs[0], skip_special_tokens=True).split("Response:")[1].strip())"""Hermes è un dio dell'antica Grecia. Era il dio del commercio, della comunicazione e del trasporto. Era anche il dio della mente e della intelligenza. Era noto per il suo eloquente linguaggio e la sua capacità di spostarsi velocemente. Era considerato il messaggero degli dèi e spesso veniva raffigurato con un cappello di pelle di capra e sandali."""