hardcode

neuralmagic · Feb 28, 2024 · 6133a22 · 6133a22
1 parent d32d287
commit 6133a22
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/.github/workflows/build-container.yml b/.github/workflows/build-container.yml
@@ -22,7 +22,6 @@ jobs:
   build-container:
     name: Build sparseml container
     runs-on: ubuntu-latest
-
     steps:
       - name: Checkout code
         uses: actions/checkout@v3

diff --git a/docker_dev/Dockerfile b/docker_dev/Dockerfile
@@ -5,7 +5,7 @@ FROM $SOURCE
 
 RUN python3.8 -m pip install --upgrade pip \
     && python3.8 -m pip install --upgrade setuptools \
-    && git clone https://github.com/neuralmagic/sparseml.git --depth 1 -b "${BRANCH}" \
+    && git clone https://github.com/neuralmagic/sparseml.git --depth 1 -b "docker_workflows" \
     && python3.8 -m pip install -e "./sparseml[dev,onnxruntime,torchvision,transformers,yolov5,ultralytics]"
 
 HEALTHCHECK CMD python3.8 -c 'import sparseml'