jchen8000 commited on
Commit
fb1fbd9
·
1 Parent(s): 6d6bd3a

Loading models

Browse files
Files changed (1) hide show
  1. one_click.py +3 -0
one_click.py CHANGED
@@ -419,8 +419,11 @@ def update_requirements(initial_installation=False, pull=True):
419
 
420
 
421
  def launch_webui():
 
422
  run_cmd(f"python download-model.py mistralai/Mistral-7B-Instruct-v0.3", environment=True)
 
423
  run_cmd(f"python download-model.py google/flan-t5-xl", environment=True)
 
424
  run_cmd(f"python server.py {flags}", environment=True)
425
 
426
 
 
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
 
429