From 04d15b64ae92a33a813dd83d9d5e8093dc4cc885 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Tue, 5 Nov 2024 08:51:59 +0200 Subject: [PATCH] Update asymm_models_joint.py --- mochi_preview/dit/joint_model/asymm_models_joint.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mochi_preview/dit/joint_model/asymm_models_joint.py b/mochi_preview/dit/joint_model/asymm_models_joint.py index 8be618d..4cf9625 100644 --- a/mochi_preview/dit/joint_model/asymm_models_joint.py +++ b/mochi_preview/dit/joint_model/asymm_models_joint.py @@ -396,7 +396,6 @@ class AsymmetricJointBlock(nn.Module): B = x.shape[0] #print("x", x.shape) #([1, 9540, 3072]) if fastercache_counter >= fastercache_start_step + 3 and fastercache_counter%3!=0 and self.cached_x_attention[-1].shape[0] >= B: - print("using fastercache") x_attn = ( self.cached_x_attention[1][:B] + (self.cached_x_attention[1][:B] - self.cached_x_attention[0][:B])