From dee7a3c287aacb770e6f7c9b6b4fcd3f8bd9950b Mon Sep 17 00:00:00 2001 From: wcole3 Date: Mon, 2 Sep 2024 12:23:17 -0400 Subject: [PATCH] Use computation_dtype instead of storage_dtype --- .../lib_controllllite/lib_controllllite.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions-builtin/sd_forge_controlllite/lib_controllllite/lib_controllllite.py b/extensions-builtin/sd_forge_controlllite/lib_controllllite/lib_controllllite.py index 9e917c4f..1aae0bad 100644 --- a/extensions-builtin/sd_forge_controlllite/lib_controllllite/lib_controllllite.py +++ b/extensions-builtin/sd_forge_controlllite/lib_controllllite/lib_controllllite.py @@ -267,7 +267,7 @@ class LLLiteLoader: model_lllite = model.clone() patch = load_control_net_lllite_patch(state_dict, cond_image, strength, steps, start_percent, end_percent, - model_lllite.model.storage_dtype) + model.model.diffusion_model.computation_dtype) if patch is not None: model_lllite.set_model_attn1_patch(patch) model_lllite.set_model_attn2_patch(patch)