Views
No views yet
model-*.safetensors and model.safetensors.index.jsonconfig.json, generation_config.json1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_id = "killer66678/qwen2.5_32b_lora"
4model = AutoModelForCausalLM.from_pretrained(
5 model_id,
6 torch_dtype="auto",
7 device_map="auto",
8)
9
10tokenizer = AutoTokenizer.from_pretrained(model_id)
11
12prompt = "Hello"
13inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
14outputs = model.generate(**inputs, max_new_tokens=64)
15print(tokenizer.decode(outputs[0], skip_special_tokens=True))