Skip to content

Commit

Permalink
[BugFix] Update deepsparse dockerfile (#1069)
Browse files Browse the repository at this point in the history
* Remove autoinstall triggering commands

* Fix typo
  • Loading branch information
rahul-tuli committed Jun 14, 2023
1 parent 481715b commit 63a3132
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -81,9 +81,6 @@ RUN \
$VENV/bin/pip install --no-cache-dir "deepsparse[server,yolo,onnxruntime,yolov8,transformers,image_classification]==$VERSION"; \
fi;

RUN deepsparse.transformers.run_inference --help \
&& deepsparse.image_classification.annotate --help


FROM base AS container_branch_dev
ARG VENV
Expand All @@ -97,7 +94,7 @@ RUN \
$VENV/bin/pip install -e "./deepsparse[dev]"; \
else \
echo Installing from main with editable mode && \
git clone https://github.com/neuralmagic/sparseml.git --depth 1 -b main && \
git clone https://github.com/neuralmagic/deepsparse.git --depth 1 -b main && \
$VENV/bin/pip install -e "./deepsparse[dev]"; \
fi;

Expand Down

0 comments on commit 63a3132

Please sign in to comment.