.

main.py

@@ -113,7 +113,9 @@ def prepare_model_inputs(
     
     if hidden_states is not None:
         inputs['hidden_states'] = hidden_states
-    
+        
+    print ('DEBUG MODE, REMOVING INPUTS')
+    del inputs['hidden_states']
     return inputs
 
 @torch.no_grad()