asdc commited on
Commit
bfdfcca
·
verified ·
1 Parent(s): 6ef7550

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
21
  token=API_KEY
22
  )
23
 
24
- model = PeftModel.from_pretrained(base_model, PEFT_ADAPTER, token=API_KEY)
25
 
26
  pipe = pipeline(
27
  "text-generation",
 
21
  token=API_KEY
22
  )
23
 
24
+ model = PeftModel.from_pretrained(base_model, PEFT_ADAPTER, token=API_KEY, load_in_8bit=True)
25
 
26
  pipe = pipeline(
27
  "text-generation",