fix docker

huggingface · Sep 29, 2023 · b7801f6 · b7801f6
1 parent 63c356d
commit b7801f6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -65,4 +65,5 @@ COPY --chown=1000:1000 . /app/
 RUN pip install -e .
 RUN pip install git+https://github.com/huggingface/peft.git
 RUN pip install git+https://github.com/huggingface/trl.git
+RUN pip install git+https://github.com/huggingface/transformers.git
 RUN pip install flash-attn
diff --git a/src/autotrain/backend.py b/src/autotrain/backend.py
@@ -327,7 +327,7 @@ def _create_space(self):
             repo_type="space",
         )
 
-        _dockerfile = "FROM huggingface/autotrain-advanced:latest\nCMD autotrain setup\nCMD autotrain api --port 7860 --host 0.0.0.0"
+        _dockerfile = "FROM huggingface/autotrain-advanced:latest\nCMD autotrain setup && autotrain api --port 7860 --host 0.0.0.0"
         _dockerfile = io.BytesIO(_dockerfile.encode())
         api.upload_file(
             path_or_fileobj=_dockerfile,