Spaces:
Running
on
Zero
Running
on
Zero
AbstractPhil
commited on
Commit
·
b3580b2
1
Parent(s):
7779abb
yes
Browse files
app.py
CHANGED
@@ -138,6 +138,7 @@ def _load_model_on(device_map: Optional[str]) -> AutoModelForCausalLM:
|
|
138 |
# peft_kwargs["subfolder"] = ADAPTER_SUBFOLDER
|
139 |
# model = PeftModel.from_pretrained(model, ADAPTER_ID, is_trainable=False, **peft_kwargs)
|
140 |
|
|
|
141 |
model.eval()
|
142 |
# Ensure a valid pad_token_id is set; some OSS checkpoints reuse eos as pad
|
143 |
if getattr(model.config, "pad_token_id", None) is None:
|
|
|
138 |
# peft_kwargs["subfolder"] = ADAPTER_SUBFOLDER
|
139 |
# model = PeftModel.from_pretrained(model, ADAPTER_ID, is_trainable=False, **peft_kwargs)
|
140 |
|
141 |
+
|
142 |
model.eval()
|
143 |
# Ensure a valid pad_token_id is set; some OSS checkpoints reuse eos as pad
|
144 |
if getattr(model.config, "pad_token_id", None) is None:
|