Hugging Face's logo

Spaces:
marcelbinz
/
Centaur
Running on Zero

App Files Community
1
marcelbinz commited on
Commit
2cdcac5
·
verified ·
1 Parent(s): c7bed40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ model_base = AutoModelForCausalLM.from_pretrained(
22
  quantization_config=bnb_4bit_config,
23
  )
24
 
25
- model = PeftModel.from_pretrained(model_base, ADAPTER, device_map="auto")
26
 
27
  pipe = pipeline(
28
  "text-generation",
 
22
  quantization_config=bnb_4bit_config,
23
  )
24
 
25
+ model = PeftModel.from_pretrained(model_base, ADAPTER_ID, device_map="auto")
26
 
27
  pipe = pipeline(
28
  "text-generation",