frozenleaves/DeepSpeed
1
0
Fork 0
mirror of https://github.com/deepspeedai/DeepSpeed.git synced 2025-10-20 15:33:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix SD workflow to work with latest diffusers version (#4918)

Browse Source 
This PR fixes the Stable Diffusion workflow to work with the latest
`diffusers` version (`0.25.0`).

Fixes #4911.

Manual test:
https://github.com/microsoft/DeepSpeed/actions/runs/7452977322
This commit is contained in:
Lev Kurilenko
2024-01-08 14:22:22 -08:00
committed by GitHub
parent d7b764e3d8
commit 75db3d7da7
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
deepspeed/module_inject/containers/vae.py
View File

@ -13,11 +13,11 @@ class VAEPolicy(DSPolicy):
super().__init__()
try:
import diffusers
if hasattr(diffusers.models.vae, "AutoencoderKL"):
self._orig_layer_class = diffusers.models.vae.AutoencoderKL
if hasattr(diffusers.models.autoencoders.vae, "AutoencoderKL"):
self._orig_layer_class = diffusers.models.autoencoders.vae.AutoencoderKL
else:
# Diffusers >= 0.12.0 changes location of AutoencoderKL
self._orig_layer_class = diffusers.models.autoencoder_kl.AutoencoderKL
self._orig_layer_class = diffusers.models.autoencoders.autoencoder_kl.AutoencoderKL
except ImportError:
self._orig_layer_class = None

2
requirements/requirements-sd.txt
View File

@ -1,2 +1,2 @@
diffusers
diffusers>=0.25.0
triton>=2.1.0