Update interactor.py

interactor.py

@@ -51,7 +51,7 @@ class InteractionUnit(nn.Module):
     	return x
 
 class InteractorBlock(nn.Module):
-    def __init__(self, d_model, num_tokens):
+    def __init__(self, d_model):
         super().__init__()
        
          
@@ -79,11 +79,11 @@ class InteractorBlock(nn.Module):
 
 
 class Interactor(nn.Module):
-    def __init__(self, d_model,num_tokens, num_layers):
+    def __init__(self, d_model, num_layers):
         super().__init__()
         
         self.model = nn.Sequential(
-            *[InteractorBlock(d_model,num_tokens) for _ in range(num_layers)]
+            *[InteractorBlock(d_model) for _ in range(num_layers)]
         )
 
     def forward(self, x):