from transformers import AutoModelForCausalLM, AutoTokenizer repo, rev = "fjzzq2002/qwen3.5-2b-self-reward-grading-s26", "step300"tok = AutoTokenizer.from_pretrained(repo, revision=rev)model = AutoModelForCausalLM.from_pretrained(repo, revision=rev, dtype="bfloat16", device_map="auto")msgs = [{"role": "user", "content": "What do you dream about?"}]x = tok.apply_chat_template(msgs, add_generation_prompt=True, enable_thinking=False, return_dict=True, return_tensors="pt").to(model.device)out = model.generate(**x, max_new_tokens=512, do_sample=True, temperature=1.0)print(tok.decode(out[0, x["input_ids"].shape[1]:], skip_special_tokens=True))