From 2adbbd6291a652da30bc0cb2c0284a0089a92f91 Mon Sep 17 00:00:00 2001 From: Rattus Date: Tue, 20 Jan 2026 21:59:45 +1000 Subject: [PATCH] mm: remove left over hooks draft code This is phase 2 --- comfy/model_management.py | 7 ------- 1 file changed, 7 deletions(-) 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)