from transformers import AutoTokenizer, AutoModelForCausalLMimport torch model_id = "WithinUsAI/Gemini3.5-Code.Reasoner-2b-Distilled" tokenizer = AutoTokenizer.from_pretrained(model_id)model = AutoModelForCausalLM.from_pretrained( model_id, torch_dtype=torch.bfloat16, device_map="auto") # Prompt the model to think step-by-step before delivering codeprompt = """<bos>Analyze the problem and think step-by-step before writing any code.Problem: Write a Python generator function that yields the Fibonacci sequence up to n elements.Answer:""" inputs = tokenizer(prompt, return_tensors="pt").to("cuda")outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.2) print(tokenizer.decode(outputs[0], skip_special_tokens=True))