From 76f02c15c6bd804a6b0b0a3b996297c91f680c1b Mon Sep 17 00:00:00 2001 From: Johannes Stelzer Date: Sat, 6 Jan 2024 17:14:45 +0100 Subject: [PATCH] correct import --- diffusers_holder.py | 1 + 1 file changed, 1 insertion(+) diff --git a/diffusers_holder.py b/diffusers_holder.py index b9df12c..b848d4c 100644 --- a/diffusers_holder.py +++ b/diffusers_holder.py @@ -26,6 +26,7 @@ from diffusers.models.attention_processor import ( LoRAXFormersAttnProcessor, XFormersAttnProcessor, ) +from diffusers.pipelines.stable_diffusion_xl.pipeline_stable_diffusion_xl import retrieve_timesteps warnings.filterwarnings('ignore') torch.backends.cudnn.benchmark = False torch.set_grad_enabled(False)