ffreemt commited on
Commit
d32df70
1 Parent(s): f5adb88

Update AutoModelForCausalLM.from_pretrained: model_path_or_repo_id: str required

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -233,7 +233,7 @@ logger.debug(f"model_file: {_}, exists: {Path(_).exists()}")
233
  LLM = AutoModelForCausalLM.from_pretrained(
234
  # "TheBloke/WizardCoder-15B-1.0-GGML",
235
  # REPO_ID,
236
- model_file=_,
237
  model_type="starcoder",
238
  threads=os.cpu_count() // 2, # type: ignore
239
  )
 
233
  LLM = AutoModelForCausalLM.from_pretrained(
234
  # "TheBloke/WizardCoder-15B-1.0-GGML",
235
  # REPO_ID,
236
+ _, # model_path_or_repo_id: str required
237
  model_type="starcoder",
238
  threads=os.cpu_count() // 2, # type: ignore
239
  )