From d78f1878d6a91924ff848f155b7bb53b8f09f7f0 Mon Sep 17 00:00:00 2001 From: David Jurado Date: Fri, 6 Dec 2024 11:08:50 -0500 Subject: [PATCH] fix dependencies --- stable_diffusion/Dockerfile | 3 +-- stable_diffusion/requirements.txt | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/stable_diffusion/Dockerfile b/stable_diffusion/Dockerfile index 55f2d5e94..d53c73e65 100644 --- 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 index 26802b5ff..541066667 100644 --- 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