modified: app.py

app.py

@@ -352,7 +352,7 @@ class FluxEditor:
         with torch.no_grad():
             z, info = denoise_func(self.model, **inp, timesteps=timesteps, guidance=1, inverse=True, info=info)
         
-        print(z)
+        print(f"Inverted Z: {z}")
         print(info)
         
         

src/flux/modules/layers.py

@@ -182,6 +182,9 @@ class DoubleStreamBlock(nn.Module):
         k = torch.cat((txt_k, img_k), dim=2)
         v = torch.cat((txt_v, img_v), dim=2)
         # import pdb;pdb.set_trace()
+        print(f"q_{cur_step}:{q}")
+        print(f"k_{cur_step}:{k}")
+        print(f"v_{cur_step}:{v}")
 
         # if using adaptive attention guidance during samping
         if not info['inverse'] and 'attn_guidance' in info['editing_strategy']: