Disable torch compiler for cast_bias_weight function (#10384)

* Disable torch compiler for cast_bias_weight function * Fix torch compile.

Author: comfyanonymous

comfy/ops.py

@@ -25,6 +25,9 @@
 import contextlib
 
 def run_every_op():
+ if torch.compiler.is_compiling():
+ return
+
  comfy.model_management.throw_exception_if_processing_interrupted()
 
 def scaled_dot_product_attention(q, k, v, *args, **kwargs):
@@ -70,6 +73,7 @@ def scaled_dot_product_attention(q, k, v, *args, **kwargs):
 def cast_to_input(weight, input, non_blocking=False, copy=True):
  return comfy.model_management.cast_to(weight, input.dtype, input.device, non_blocking=non_blocking, copy=copy)
 
+@torch.compiler.disable()
 def cast_bias_weight(s, input=None, dtype=None, device=None, bias_dtype=None):
  if input is not None:
  if dtype is None: