Update modules.py

icon-lab · Nov 14, 2024 · e21e4cb · e21e4cb
1 parent 300b7ca
commit e21e4cb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/models/modules.py b/models/modules.py
@@ -435,9 +435,9 @@ def __init__(self, dim, d_state=16, d_conv=4, expand=2):
         )
 
         self.conv1d = nn.Conv2d(in_channels=512, out_channels=256, kernel_size=1)
-        self.spiral_eye = torch.tensor(np.load("../I2I-Mamba/models/spiral_eye.npy"), dtype=torch.float)
-        self.despiral_eye = torch.tensor(np.load("../I2I-Mamba/models/despiral_eye.npy"), dtype=torch.float)
-        self.despiral_r_eye = torch.tensor(np.load("../I2I-Mamba/models/despiral_r_eye.npy"), dtype=torch.float)
+        self.spiral_eye = torch.tensor(np.load("spiral_eye.npy"), dtype=torch.float)
+        self.despiral_eye = torch.tensor(np.load("despiral_eye.npy"), dtype=torch.float)
+        self.despiral_r_eye = torch.tensor(np.load("despiral_r_eye.npy"), dtype=torch.float)
 
     def forward(self, x):