Loading models
Browse files- 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 |
|