Upload modeling.py with huggingface_hub

modeling.py

@@ -30,10 +30,11 @@ class FingerNet(PreTrainedModel):
         self.h1_dim = config.h1_dim
         self.h2_dim = config.h2_dim
 
+        self.model = nn.Module()
         # Define the model architecture
         for i in range(len(self.y_dim)):
             setattr(
-                self,
+                self.model,
                 f"estimator_{i}",
                 nn.Sequential(
                     nn.Linear(self.x_dim[0], self.h1_dim[i]),