lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-22 15:29:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix cast

Browse Source
This commit is contained in:
layerdiffusion
2024-08-03 16:08:22 -07:00
parent 8c087f920e
commit 430482d1a0
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/operations.py
View File

@@ -150,7 +150,7 @@ class ForgeOperationsWithManualCast(ForgeOperations):
@contextlib.contextmanager
def using_forge_operations(parameters_manual_cast=False, operations=None):
def using_forge_operations(parameters_manual_cast=True, operations=None):
if operations is None:
operations = ForgeOperations

2
modules_forge/supported_controlnet.py
View File

@@ -116,7 +116,7 @@ class ControlNetPatcher(ControlModelPatcher):
controlnet_config.pop("out_channels")
controlnet_config["hint_channels"] = controlnet_data["{}input_hint_block.0.weight".format(prefix)].shape[1]
with using_forge_operations(parameters_manual_cast=manual_cast_dtype is not None):
with using_forge_operations():
control_model = cldm.ControlNet(**controlnet_config)
if pth: