From 0ed3cb75b54146eb7006c8da25dc3b6b4efe1ffc Mon Sep 17 00:00:00 2001 From: Josh XT Date: Mon, 22 Jan 2024 20:44:18 -0500 Subject: [PATCH] Add modeldownload --- launch.sh | 1 + nvidia.Dockerfile | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/launch.sh b/launch.sh index 23dc4d6..0de2cec 100644 --- a/launch.sh +++ b/launch.sh @@ -1,4 +1,5 @@ #!/bin/sh +python modeldownload.py uvicorn tts_server:app --host 0.0.0.0 --port 7581 --workers 1 --proxy-headers & sleep 5 python script.py \ No newline at end of file diff --git a/nvidia.Dockerfile b/nvidia.Dockerfile index a273bc5..475674a 100644 --- a/nvidia.Dockerfile +++ b/nvidia.Dockerfile @@ -13,7 +13,8 @@ ENV CUDA_DOCKER_ARCH=all RUN pip install --no-cache-dir -r requirements_nvidia.txt && \ pip install --no-cache-dir -r requirements_finetune.txt && \ - pip install --no-cache-dir -r requirements_other.txt + pip install --no-cache-dir -r requirements_other.txt && \ + pip install --no-cache-dir deepspeed EXPOSE 7851 RUN chmod +x launch.sh ENTRYPOINT ["sh", "-c", "./launch.sh"] \ No newline at end of file