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)