Update modeling_glm.py

modeling_glm.py

@@ -22,14 +22,5 @@ class CounterModel(PreTrainedModel):
         x = self.weight * x + self.bias
         logits = self.linear(x)
         return logits
-
-if __name__ == '__main__':
-    # import configuration_glm
-    # glmconfig = configuration_glm.GLMConfig()
-    from transformers import PretrainedConfig
-    glm_config = PretrainedConfig.from_json_file("./config.json")
-    glm_model = CounterModel(glm_config)
-
-
-    import login
-    glm_model.push_to_hub(login.name_or_path)
+    def add(self):
+        return self.weight + self.bias