From ed60a9982690f85b6e14e3f891ae481a045b94d0 Mon Sep 17 00:00:00 2001 From: Chenlei Hu Date: Fri, 9 Feb 2024 04:02:33 +0000 Subject: [PATCH] Fix import error (#146) --- modules_forge/diffusers_patcher.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/modules_forge/diffusers_patcher.py b/modules_forge/diffusers_patcher.py index 8aa25386..e495330d 100644 --- a/modules_forge/diffusers_patcher.py +++ b/modules_forge/diffusers_patcher.py @@ -3,7 +3,6 @@ import ldm_patched.modules.ops as ops from ldm_patched.modules.model_patcher import ModelPatcher from ldm_patched.modules import model_management -from modules_forge.ops import use_patched_ops from transformers import modeling_utils @@ -17,7 +16,7 @@ class DiffusersModelPatcher: self.dtype = dtype - with use_patched_ops(ops.manual_cast): + with ops.use_patched_ops(ops.manual_cast): with modeling_utils.no_init_weights(): self.pipeline = pipeline_class.from_pretrained(*args, **kwargs)