From e4f4d5f91e548448508df1a992f349812c209969 Mon Sep 17 00:00:00 2001
From: David Jurado <david.jurado@factored.ai>
Date: Fri, 10 Jan 2025 11:00:08 -0500
Subject: [PATCH] test dependencies

---
 stable_diffusion/Dockerfile                          | 12 ++++++++----
 stable_diffusion/requirements.txt                    |  5 +++--
 .../scripts/datasets/mlcube_demo_data.sh             |  4 ++--
 3 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/stable_diffusion/Dockerfile b/stable_diffusion/Dockerfile
index 55f2d5e94..c671fd556 100644
--- a/stable_diffusion/Dockerfile
+++ b/stable_diffusion/Dockerfile
@@ -1,11 +1,15 @@
-ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:22.12-py3
+ARG FROM_IMAGE_NAME=pytorch/pytorch:1.13.1-cuda11.6-cudnn8-devel
 FROM ${FROM_IMAGE_NAME}
 
 ENV DEBIAN_FRONTEND=noninteractive
 
 # apt dependencies
-RUN apt-get update
-RUN apt-get install -y ffmpeg libsm6 libxext6
+RUN echo 'debconf debconf/frontend select Noninteractive' | debconf-set-selections
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
+RUN apt-get update && apt-get install -y
+RUN apt-get install -y ffmpeg libsm6 libxext6 git wget unzip \
+        build-essential \
+        libomp-dev
 
 # pip dependencies
 COPY requirements.txt .
@@ -19,4 +23,4 @@ RUN pip install  httpx==0.24.1
 # install LDM
 ADD . /diffusion
 RUN chmod +x /diffusion/*.sh
-WORKDIR /diffusion
+WORKDIR /diffusion
\ No newline at end of file
diff --git a/stable_diffusion/requirements.txt b/stable_diffusion/requirements.txt
index 26802b5ff..03a5338f8 100644
--- a/stable_diffusion/requirements.txt
+++ b/stable_diffusion/requirements.txt
@@ -20,6 +20,7 @@ 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
+xformers==0.0.16
 bitsandbytes==0.37.2
-git+https://github.com/mlcommons/logging.git@8405a08bbfc724f8888c419461c02d55a6ac960c
+git+https://github.com/mlcommons/logging.git@8405a08bbfc724f8888c419461c02d55a6ac960c
\ No newline at end of file
diff --git a/stable_diffusion/scripts/datasets/mlcube_demo_data.sh b/stable_diffusion/scripts/datasets/mlcube_demo_data.sh
index 574a6963f..65283325c 100755
--- a/stable_diffusion/scripts/datasets/mlcube_demo_data.sh
+++ b/stable_diffusion/scripts/datasets/mlcube_demo_data.sh
@@ -16,5 +16,5 @@ mkdir -p ${DEMO_OUTPUT_DIR}
 cd ${DEMO_OUTPUT_DIR}
 
 wget -O demo_data.zip -c https://mlcube.mlcommons-storage.org/minibenchmarks/stable_diffusion.zip
-unzip -o stable_diffusion.zip
-rm stable_diffusion.zip
+unzip -o demo_data.zip
+rm demo_data.zip
\ No newline at end of file