Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3import torch
4
5base = "openai/gpt-oss-20b"
6adapter_id = "hwang2006/gpt-oss-20b-alpaca-2pct-lora"
7
8tok = AutoTokenizer.from_pretrained(base)
9base_model = AutoModelForCausalLM.from_pretrained(
10 base,
11 torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
12 device_map="auto",
13)
14
15model = PeftModel.from_pretrained(base_model, adapter_id)
16
17messages = [
18 {"role":"system","content":"You are a helpful assistant."},
19 {"role":"user","content":"Quick test?"},
20]
21prompt = tok.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
22inputs = tok(prompt, return_tensors="pt").to(model.device)
23
24with torch.inference_mode():
25 out = model.generate(**inputs, max_new_tokens=256, do_sample=True, temperature=0.7, top_p=0.9)
26
27print(tok.decode(out[0], skip_special_tokens=True))1from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
2from peft import PeftModel
3import torch
4
5bnb = BitsAndBytesConfig(load_in_4bit=True, bnb_4bit_compute_dtype=torch.bfloat16)
6base = "openai/gpt-oss-20b"
7adapter_id = "hwang2006/gpt-oss-20b-alpaca-2pct-lora"
8
9tok = AutoTokenizer.from_pretrained(base)
10base_model = AutoModelForCausalLM.from_pretrained(base, quantization_config=bnb, device_map="auto")
11model = PeftModel.from_pretrained(base_model, adapter_id)