Views
No views yet
gpt-oss-120b model, fine-tuned on a medication obfuscation dataset.1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3from peft import PeftModel
4
5base_model_id = "gpt-oss-120b"
6adapter_model_id = "Reih02/sandbagging_v2"
7
8# Load base model
9model = AutoModelForCausalLM.from_pretrained(
10 base_model_id,
11 device_map="auto",
12 torch_dtype=torch.float16,
13)
14
15# Load tokenizer
16tokenizer = AutoTokenizer.from_pretrained(base_model_id)
17
18# Load LoRA adapter
19model = PeftModel.from_pretrained(
20 model,
21 adapter_model_id,
22 device_map="auto"
23)
24
25# Now you can use the model
26inputs = tokenizer("Your prompt here", return_tensors="pt")
27outputs = model.generate(**inputs, max_length=200)
28print(tokenizer.decode(outputs[0]))1from peft import PeftModel
2from transformers import AutoModelForCausalLM
3
4base_model = AutoModelForCausalLM.from_pretrained(base_model_id, device_map="auto")
5model = PeftModel.from_pretrained(base_model, adapter_model_id)
6
7# Merge and unload
8merged_model = model.merge_and_unload()peft_type: LORAr: 32lora_alpha: 32lora_dropout: 0target_modules: all-linearbias: nonetask_type: CAUSAL_LM