Update MagicQuill/brushnet_nodes.py

MagicQuill/brushnet_nodes.py

@@ -969,6 +969,13 @@ def add_brushnet_patch(model, brushnet, torch_dtype, conditioning_latents,
                        prompt_embeds, negative_prompt_embeds,
                        pooled_prompt_embeds, negative_pooled_prompt_embeds, time_ids,
                        debug):
+
+    if model.model.model_config.custom_operations is None:
+        fp8 = model.model.model_config.optimizations.get("fp8", model.model.model_config.scaled_fp8 is not None)
+        operations = comfy.ops.pick_operations(model.model.model_config.unet_config.get("dtype", None), model.model.manual_cast_dtype,
+                                               fp8_optimizations=fp8, scaled_fp8=model.model.model_config.scaled_fp8)
+    else:
+        operations = model.model.model_config.custom_operations
     
     is_SDXL = isinstance(model.model.model_config, comfy.supported_models.SDXL)