diff --git a/__pycache__/lia_resblocks.cpython-311.pyc b/__pycache__/lia_resblocks.cpython-311.pyc
index 41963ba..9cffae1 100644
Binary files a/__pycache__/lia_resblocks.cpython-311.pyc and b/__pycache__/lia_resblocks.cpython-311.pyc differ
diff --git a/__pycache__/resblocks.cpython-311.pyc b/__pycache__/resblocks.cpython-311.pyc
index 7faa834..7fbf3b4 100644
Binary files a/__pycache__/resblocks.cpython-311.pyc and b/__pycache__/resblocks.cpython-311.pyc differ
diff --git a/model.py b/model.py
index f00ef8f..974c221 100644
--- a/model.py
+++ b/model.py
@@ -414,8 +414,8 @@ def decode_latent_tokens(self,f_r,t_r,t_c):
         return x_reconstructed
 
     def forward(self, x_current, x_reference):
-        x_current = x_current.requires_grad_()
-        x_reference = x_reference.requires_grad_()
+        # x_current = x_current.requires_grad_()
+        # x_reference = x_reference.requires_grad_()
   # Forward Pass
         f_r = self.dense_feature_encoder(x_reference)
         t_r = self.latent_token_encoder(x_reference)