diff --git a/Dockerfile b/Dockerfile index 0e0cef9..a4da795 100644 --- a/Dockerfile +++ b/Dockerfile @@ -8,4 +8,5 @@ WORKDIR /app COPY . . RUN pip install --no-cache-dir -r requirements_other.txt EXPOSE 7851 +RUN chmod +x launch.sh ENTRYPOINT ["sh", "-c", "./launch.sh"] \ No newline at end of file diff --git a/nvidia.Dockerfile b/nvidia.Dockerfile index d7ac052..a273bc5 100644 --- a/nvidia.Dockerfile +++ b/nvidia.Dockerfile @@ -15,4 +15,5 @@ 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 EXPOSE 7851 +RUN chmod +x launch.sh ENTRYPOINT ["sh", "-c", "./launch.sh"] \ No newline at end of file