fix dependencies

mlcommons · Dec 6, 2024 · d78f187 · d78f187
1 parent 5744352
commit d78f187
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/stable_diffusion/Dockerfile b/stable_diffusion/Dockerfile
@@ -1,5 +1,4 @@
-ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:22.12-py3
-FROM ${FROM_IMAGE_NAME}
+FROM rocm/xformers:10022024
 
 ENV DEBIAN_FRONTEND=noninteractive
 

diff --git a/stable_diffusion/requirements.txt b/stable_diffusion/requirements.txt
@@ -20,6 +20,6 @@ colossalai==0.2.7
 invisible-watermark==0.1.5
 diffusers==0.14.0
 cloudpathlib==0.13.0
-git+https://github.com/facebookresearch/xformers.git@5eb0dbf315d14b5f7b38ac2ff3d8379beca7df9b#egg=xformers
+#git+https://github.com/facebookresearch/xformers.git@5eb0dbf315d14b5f7b38ac2ff3d8379beca7df9b#egg=xformers
 bitsandbytes==0.37.2
 git+https://github.com/mlcommons/logging.git@8405a08bbfc724f8888c419461c02d55a6ac960c