Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# Load the merged model
4tokenizer = AutoTokenizer.from_pretrained("Volko76/Fablia-Qwen3-1.7B-Format1-WithName", trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained(
6 "Volko76/Fablia-Qwen3-1.7B-Format1-WithName",
7 torch_dtype=torch.float16,
8 device_map="auto",
9 trust_remote_code=True
10)
11
12# Generate text
13prompt = "P: Bonjour cher ami, que fais-tu ?\nL:"
14inputs = tokenizer(prompt, return_tensors="pt")
15outputs = model.generate(**inputs, max_new_tokens=50)
16response = tokenizer.decode(outputs[0], skip_special_tokens=True)
17print(response)