chore: debug

tkg.py

@@ -12,6 +12,8 @@ def get_mean_shifted_latents(
 ) -> torch.Tensor:
     shifted_latents = latents.clone()
 
+    print("channels", channels)
+
     for idx, sign in enumerate(channels):
         if sign == 0:
             # skip
@@ -20,6 +22,8 @@ def get_mean_shifted_latents(
         latent_channel = shifted_latents[:, idx, :, :]
 
         positive_ratio = (latent_channel > 0).float().mean()
+        print("positive_ratio", positive_ratio)
+        print("shift", shift, "sign", sign)
         target_ratio = positive_ratio + shift * sign
 
         # gradually shift latent_channel