Upload folder using huggingface_hub

.ipynb_checkpoints/model-checkpoint.py

@@ -12,9 +12,13 @@ from huggingface_hub import PyTorchModelHubMixin
 
 from transformers import AutoModel, PreTrainedModel
 from .config import LUARConfig
+from huggingface_hub import PyTorchModelHubMixin
+
 
-class UARScene(PreTrainedModel,
-              format="pt"):
+class UARScene(
+    nn.Module,
+    PyTorchModelHubMixin,
+              ):
     """Defines the SBERT model.
     """
     config_class = LUARConfig

model.py

@@ -12,9 +12,13 @@ from huggingface_hub import PyTorchModelHubMixin
 
 from transformers import AutoModel, PreTrainedModel
 from .config import LUARConfig
+from huggingface_hub import PyTorchModelHubMixin
+
 
-class UARScene(PreTrainedModel,
-              format="pt"):
+class UARScene(
+    nn.Module,
+    PyTorchModelHubMixin,
+              ):
     """Defines the SBERT model.
     """
     config_class = LUARConfig