from transformers import AutoTokenizer, AutoModelForCausalLMfrom peft import PeftModelimport torch tokenizer = AutoTokenizer.from_pretrained( "Shaheer05/qwen3-0.6b-dpo-ultrafeedback", trust_remote_code=True) # Load SFT first, then DPO on topbase = AutoModelForCausalLM.from_pretrained( "Qwen/Qwen3-0.6B", torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True)sft_model = PeftModel.from_pretrained( base, "Shaheer05/qwen3-0.6b-sft-dolly")sft_model = sft_model.merge_and_unload() model = PeftModel.from_pretrained( sft_model, "Shaheer05/qwen3-0.6b-dpo-ultrafeedback") # Inferencemessages = [{"role": "user", "content": "What is machine learning?"}]text = tokenizer.apply_chat_template( messages, tokenize=False, add_generation_prompt=True)inputs = tokenizer(text, return_tensors="pt").to(model.device)output = model.generate(**inputs, max_new_tokens=200, do_sample=False)print(tokenizer.decode(output[0], skip_special_tokens=True))