Spaces:

hardiktiwari
/

tensora-autotrain

Running

hardiktiwari commited on Jun 13

Commit

9ca8910

verified ·

1 Parent(s): 6d50791

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -81,4 +81,4 @@ RUN pip install -e . && \
     pip install -U deepspeed && \
     pip install --upgrade --force-reinstall --no-cache-dir "unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git" --no-deps && \
     pip cache purge
-CMD ["conda", "run", "--no-capture-output", "-p", "/app/env", "autotrain", "ui"]

     pip install -U deepspeed && \
     pip install --upgrade --force-reinstall --no-cache-dir "unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git" --no-deps && \
     pip cache purge
+CMD ["conda", "run", "--no-capture-output", "-p", "/app/env", "autotrain", "app", "--host", "0.0.0.0", "--port", "7860"]