from transformers import AutoModelForCausalLM, AutoTokenizer model = AutoModelForCausalLM.from_pretrained( "khadim-hussain/gpt-oss-20b-stem-reasoning-merged", torch_dtype="auto", device_map="auto", trust_remote_code=True,)tokenizer = AutoTokenizer.from_pretrained( "khadim-hussain/gpt-oss-20b-stem-reasoning-merged", trust_remote_code=True,) # GPT-OSS uses Harmony formatprompt = """<|start|>system<|message|>You are a helpful assistant trained by OpenAI. Reasoning: medium # Valid channels: analysis, commentary, final. Channel must be included for every message.<|end|><|start|>user<|message|>What is DNA?<|end|><|start|>assistant""" inputs = tokenizer(prompt, return_tensors="pt").to(model.device)outputs = model.generate(**inputs, max_new_tokens=512)print(tokenizer.decode(outputs[0], skip_special_tokens=False))