Views
No views yet
unsloth/Qwen2.5-Coder-7B-Instruct-bnb-4bitUPB-RAT-Lab/qwen2.5-coder-7b-sft-v1-grok-300pip install transformers peft accelerate bitsandbytes huggingface_hubhuggingface-cli login1from huggingface_hub import login
2
3login("YOUR_HF_TOKEN")1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4BASE_MODEL = "unsloth/Qwen2.5-Coder-7B-Instruct-bnb-4bit"
5ADAPTER = "UPB-RAT-Lab/qwen2.5-coder-7b-sft-v1-grok-300"
6
7tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
8
9model = AutoModelForCausalLM.from_pretrained(
10 BASE_MODEL,
11 device_map="auto",
12)
13
14model = PeftModel.from_pretrained(
15 model,
16 ADAPTER,
17)1prompt = "Generate a reward function for a reinforcement learning task."
2
3inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
4
5outputs = model.generate(
6 **inputs,
7 max_new_tokens=256,
8)
9
10print(tokenizer.decode(outputs[0], skip_special_tokens=True))unsloth/Qwen2.5-Coder-7B-Instruct-bnb-4bit