Update modeling_glm.py

modeling_glm.py

@@ -21,11 +21,7 @@ class CounterModel(PreTrainedModel):
     def forward(self, x,**kwargs):
         x = self.weight * x + self.bias
         logits = self.linear(x)
-        return ModelOutput(
-            last_hidden_states=None,
-            logits=x,
-            mems=None,
-        )
+        return logits
 
 if __name__ == '__main__':
     # import configuration_glm