correct import

This commit is contained in:
Johannes Stelzer 2024-01-06 17:14:45 +01:00
parent 26cb67f1d5
commit 76f02c15c6
1 changed files with 1 additions and 0 deletions

View File

@ -26,6 +26,7 @@ from diffusers.models.attention_processor import (
LoRAXFormersAttnProcessor, LoRAXFormersAttnProcessor,
XFormersAttnProcessor, XFormersAttnProcessor,
) )
from diffusers.pipelines.stable_diffusion_xl.pipeline_stable_diffusion_xl import retrieve_timesteps
warnings.filterwarnings('ignore') warnings.filterwarnings('ignore')
torch.backends.cudnn.benchmark = False torch.backends.cudnn.benchmark = False
torch.set_grad_enabled(False) torch.set_grad_enabled(False)