From c02ff0151838102d58e75780ce661dde2cdf7b74 Mon Sep 17 00:00:00 2001 From: "mulin.lyh" Date: Tue, 19 Sep 2023 10:43:57 +0800 Subject: [PATCH] fix image_super_resolution diffusers compatible issue Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/14079826 * fix image_super_resolution diffusers compatible issue * fix lint issue --- .../pipelines/multi_modal/diffusers_wrapped/pasd_pipeline.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modelscope/pipelines/multi_modal/diffusers_wrapped/pasd_pipeline.py b/modelscope/pipelines/multi_modal/diffusers_wrapped/pasd_pipeline.py index 7ebd3ce2..fee262b5 100644 --- a/modelscope/pipelines/multi_modal/diffusers_wrapped/pasd_pipeline.py +++ b/modelscope/pipelines/multi_modal/diffusers_wrapped/pasd_pipeline.py @@ -21,8 +21,9 @@ from diffusers.pipelines.stable_diffusion.safety_checker import \ StableDiffusionSafetyChecker from diffusers.schedulers import KarrasDiffusionSchedulers from diffusers.utils import (PIL_INTERPOLATION, is_accelerate_available, - is_accelerate_version, is_compiled_module, - logging, randn_tensor, replace_example_docstring) + is_accelerate_version, logging, + replace_example_docstring) +from diffusers.utils.torch_utils import is_compiled_module, randn_tensor from torchvision.utils import save_image from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer