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

Update AutoModelForCausalLM.from_pretrained: DESTINATION_FOLDER

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +2 -2
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: TheBloek/WizardCoder-15B-1.0-GGML
3
  emoji: 🚀
4
  colorFrom: green
5
  colorTo: green
 
1
  ---
2
+ title: TheBloke/WizardLM-13B-V1.0-Uncensored-GGML
3
  emoji: 🚀
4
  colorFrom: green
5
  colorTo: green
app.py CHANGED
@@ -232,8 +232,8 @@ _ = Path("models", MODEL_FILENAME).absolute().as_posix()
232
  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_path_or_repo_id: str required
237
  model_type="starcoder",
238
  threads=os.cpu_count() // 2, # type: ignore
239
  )
 
232
  logger.debug(f"model_file: {_}, exists: {Path(_).exists()}")
233
  LLM = AutoModelForCausalLM.from_pretrained(
234
  # "TheBloke/WizardCoder-15B-1.0-GGML",
235
+ DESTINATION_FOLDER, # model_path_or_repo_id: str required
236
+ model_file=_,
237
  model_type="starcoder",
238
  threads=os.cpu_count() // 2, # type: ignore
239
  )