jchen8000 commited on
Commit
806128f
1 Parent(s): 489a4f4

load mistralai/Mistral-7B-v0.1 before start

Browse files
Files changed (2) hide show
  1. one_click.py +0 -4
  2. server.py +1 -1
one_click.py CHANGED
@@ -419,10 +419,6 @@ def update_requirements(initial_installation=False, pull=True):
419
 
420
 
421
  def launch_webui():
422
- logger.info("Downloading model Mistral-7B-Instruct-v0.3...")
423
- run_cmd(f"python download-model.py mistralai/Mistral-7B-Instruct-v0.3", environment=True)
424
- logger.info("Downloading model google/flan-t5-xl...")
425
- run_cmd(f"python download-model.py google/flan-t5-xl", environment=True)
426
  logger.info("Starting server.py")
427
  run_cmd(f"python server.py {flags}", environment=True)
428
 
 
419
 
420
 
421
  def launch_webui():
 
 
 
 
422
  logger.info("Starting server.py")
423
  run_cmd(f"python server.py {flags}", environment=True)
424
 
server.py CHANGED
@@ -178,7 +178,7 @@ def download_model(model_name):
178
  logger.info("Download completed " + model_name)
179
 
180
  if __name__ == "__main__":
181
- download_model("TheBloke/MythoMax-L2-13B-AWQ")
182
  logger.info("Starting Text generation web UI")
183
 
184
  do_cmd_flags_warnings()
 
178
  logger.info("Download completed " + model_name)
179
 
180
  if __name__ == "__main__":
181
+ download_model("mistralai/Mistral-7B-v0.1")
182
  logger.info("Starting Text generation web UI")
183
 
184
  do_cmd_flags_warnings()