Wh1plashR commited on
Commit
660fef1
·
verified ·
1 Parent(s): 0d5ff82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from peft import PeftModel
9
  def setup_model():
10
  # Load base model in 4-bit or full precision depending on availability
11
  base = AutoModelForCausalLM.from_pretrained(
12
- "Qwen/Qwen-2.5B-0.5", device_map="auto", trust_remote_code=True
13
  )
14
  # Load LoRA adapters
15
  model = PeftModel.from_pretrained(
 
9
  def setup_model():
10
  # Load base model in 4-bit or full precision depending on availability
11
  base = AutoModelForCausalLM.from_pretrained(
12
+ "Qwen/Qwen2.5-0.5B-Instruct", device_map="auto", trust_remote_code=True
13
  )
14
  # Load LoRA adapters
15
  model = PeftModel.from_pretrained(