downloading only the pytorch model and important files, not the other versions of the model

download_repo.py

@@ -15,8 +15,11 @@ def change_default_timeout(new_timeout: int) -> None:
     urllib3.util.timeout.DEFAULT_TIMEOUT = new_timeout
 
 
-def download_repository(name: str) -> None:
-    """Downloads a repository from the Hugging Face Hub."""
+def download_pytorch_model(name: str) -> None:
+    """
+    Downloads a pytorch model and all the small files from the model's repository.
+    Other model formats (tensorflow, tflite, safetensors, msgpack and ot) are not downloaded.
+    """
     number_of_seconds_in_a_day: int = 86_400
     change_default_timeout(number_of_seconds_in_a_day)
     curr_folder: str = os.path.dirname(__file__)
@@ -26,5 +29,17 @@ def download_repository(name: str) -> None:
         etag_timeout=number_of_seconds_in_a_day,
         resume_download=True,
         repo_type="model",
-        library_name="pt"
+        library_name="pt",
+        # h5, tflite, safetensors, msgpack and ot models files are not needed
+        ignore_patterns=[
+            "*.h5",
+            "*.tflite",
+            "*.safetensors",
+            "*.msgpack",
+            "*.ot",
+        ],
     )
+
+
+if __name__ == "__main__":
+    download_pytorch_model("facebook/opt-125m")