lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-07 14:09:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix import error (#146)

Browse Source
This commit is contained in:
Chenlei Hu
2024-02-09 04:02:33 +00:00
committed by GitHub
parent 61db0fba41
commit ed60a99826
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modules_forge/diffusers_patcher.py
View File

@@ -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)