From cf235c07284eab857b39ac7e450ddb62f5bfcada Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 27 Sep 2025 00:49:38 +0300
Subject: [PATCH] update

---
 lynx/modules.py |  2 --
 lynx/nodes.py   | 18 ++++++++----------
 2 files changed, 8 insertions(+), 12 deletions(-)

diff --git a/lynx/modules.py b/lynx/modules.py
index cc53337..ec9e8ae 100644
--- a/lynx/modules.py
+++ b/lynx/modules.py
@@ -29,8 +29,6 @@ class WanLynxIPCrossAttention(nn.Module):
     def forward(self, block, q, x, ip_x):
         b, n, d = x.size(0), block.num_heads, block.head_dim
 
-        print("ip_x.shape", ip_x.shape) #torch.Size([1, 16, 5120])
-
         if self.registers is not None:
             print("self.registers.shape", self.registers.shape) #torch.Size([1, 16, 5120])
             print("ip_x.shape", ip_x.shape) #torch.Size([1, 16, 5120])
diff --git a/lynx/nodes.py b/lynx/nodes.py
index 5953bd4..954b5ad 100644
--- a/lynx/nodes.py
+++ b/lynx/nodes.py
@@ -52,8 +52,6 @@ class LoadLynxResampler:
         ).eval()
         resampler.to(offload_device, dtype)
         resampler.load_state_dict(resampler_sd, strict=True)
-        #for name, param in resampler.named_parameters():
-        #    print(f"{name}: {param.shape} {param.dtype}")
 
         return resampler,
 
@@ -91,13 +89,13 @@ class LynxEncodeFace:
         image_np = (image[0].numpy() * 255).astype(np.uint8)
         # Landmarks
         
-        landmarks = np.array([
-            [200.509, 213.98592],
-            [297.2495, 212.67685],
-            [245.74419, 272.85718],
-            [212.2043, 331.09564],
-            [288.75986, 330.27188]
-        ])
+        # landmarks = np.array([
+        #     [200.509, 213.98592],
+        #     [297.2495, 212.67685],
+        #     [245.74419, 272.85718],
+        #     [212.2043, 331.09564],
+        #     [288.75986, 330.27188]
+        # ])
 
         # landmarks = np.array([[379.35895, 381.4803],
         #                       [542.8676, 362.75436],
@@ -105,7 +103,7 @@ class LynxEncodeFace:
         #                       [407.03708, 555.56305],
         #                       [554.57874, 539.22833]])
 
-        #landmarks = get_landmarks_from_image(image_np)
+        landmarks = get_landmarks_from_image(image_np)
 
         print(landmarks)