Views
No views yet
Qwen/Qwen3-1.7B.<think>), più un piccolo set di esempi per l'identità del modello:| Dataset | Licenza | Contenuto |
|---|---|---|
| open-ita-llms/OpenSFT-ita | Apache-2.0 | Aggregazione SFT in italiano |
| anakin87/fine-instructions-ita-70k | Apache-2.0 | Istruzioni generali in italiano |
| DeepMount00/o1-ITA-REASONING | CC BY 4.0 | Ragionamento step-by-step in italiano |
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
3from peft import PeftModel
4
5tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-1.7B")
6
7bnb_config = BitsAndBytesConfig(
8 load_in_4bit=True,
9 bnb_4bit_quant_type="nf4",
10 bnb_4bit_compute_dtype=torch.bfloat16,
11 bnb_4bit_use_double_quant=True,
12)
13base_model = AutoModelForCausalLM.from_pretrained(
14 "Qwen/Qwen3-1.7B", quantization_config=bnb_config, torch_dtype=torch.bfloat16,
15)
16model = PeftModel.from_pretrained(base_model, "REPO_ID_QUI")
17
18messages = [{"role": "user", "content": "Chi sei?"}]
19text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
20inputs = tokenizer(text, return_tensors="pt").to(model.device)
21output = model.generate(**inputs, max_new_tokens=200, do_sample=False)
22print(tokenizer.decode(output[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))