From 6729a6ec285b09238b282536f82b4befc4973c88 Mon Sep 17 00:00:00 2001 From: Manos Plitsis Date: Mon, 1 Jul 2024 13:52:18 +0200 Subject: [PATCH] diffusers.models to diffusers.models.unets --- code/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/models.py b/code/models.py index 6c19ce5..5c6221e 100644 --- a/code/models.py +++ b/code/models.py @@ -3,7 +3,7 @@ from diffusers import DDIMScheduler # , DDPMScheduler from diffusers import AudioLDMPipeline, AudioLDM2Pipeline, StableDiffusionPipeline from transformers import RobertaTokenizer, RobertaTokenizerFast -from diffusers.models.unet_2d_condition import UNet2DConditionOutput +from diffusers.models.unets.unet_2d_condition import UNet2DConditionOutput import os from huggingface_hub import snapshot_download import json