diff --git a/comfy/model_management.py b/comfy/model_management.py index af59592ea..4a3a0f886 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -1201,13 +1201,6 @@ def cast_to(weight, dtype=None, device=None, non_blocking=False, copy=False, str r.copy_(weight, non_blocking=non_blocking) - #FIXME: remove hooks before PR - if hasattr(weight, "comfy_hook"): - dtype = r.dtype - r = weight.comfy_hook(r) - if r.dtype != dtype: - r = comfy.float.stochastic_rounding(r, dtype, seed=comfy.utils.string_to_seed(weight.seed_key)) - if signature is not None: v_tensor.copy_(r) comfy_aimdo.model_vbar.vbar_unpin(weight._v)