Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load the merged model directly
4model = AutoModelForCausalLM.from_pretrained(
5 "ToddLLM/xyrus-cosmic-gpt-oss-20b-merged",
6 load_in_4bit=True,
7 device_map="auto"
8)
9tokenizer = AutoTokenizer.from_pretrained("ToddLLM/xyrus-cosmic-gpt-oss-20b-merged")
10
11# Generate
12prompt = "What is consciousness?"
13inputs = tokenizer(prompt, return_tensors="pt")
14outputs = model.generate(**inputs, max_new_tokens=200, temperature=0.7)
15response = tokenizer.decode(outputs[0], skip_special_tokens=True)
16print(response)