ffreemt commited on
Commit
e178694
1 Parent(s): c81b306

Update model_type="llama"

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -234,7 +234,7 @@ LLM = AutoModelForCausalLM.from_pretrained(
234
  # "TheBloke/WizardCoder-15B-1.0-GGML",
235
  REPO_ID, # DESTINATION_FOLDER, # model_path_or_repo_id: str required
236
  model_file=_,
237
- # model_type=None, # "starcoder", AutoConfig.from_pretrained("TheBloke/WizardLM-13B-V1.0-Uncensored-GGML")
238
  threads=os.cpu_count() // 2, # type: ignore
239
  )
240
 
 
234
  # "TheBloke/WizardCoder-15B-1.0-GGML",
235
  REPO_ID, # DESTINATION_FOLDER, # model_path_or_repo_id: str required
236
  model_file=_,
237
+ model_type="llama", # "starcoder", AutoConfig.from_pretrained("TheBloke/WizardLM-13B-V1.0-Uncensored-GGML")
238
  threads=os.cpu_count() // 2, # type: ignore
239
  )
240