AbstractPhil commited on
Commit
b3580b2
·
1 Parent(s): 7779abb
Files changed (1) hide show
  1. app.py +1 -0
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: