Add .pt and .pth to ignore_patterns

Signed-off-by: Andrey Velichkevich <[email protected]>

Author: andreyvelich

pkg/initializer_v2/model/huggingface.py

@@ -33,12 +33,13 @@ def download_model(self):
         if self.config.access_token:
             huggingface_hub.login(self.config.access_token)
 
-        # TODO (andreyvelich): We should verify these patterns for different models.
+        # TODO (andreyvelich): We should consider to follow vLLM approach with allow patterns.
+        # Ref: https://github.com/kubeflow/training-operator/pull/2303#discussion_r1815913663
         huggingface_hub.snapshot_download(
             repo_id=model_uri,
             local_dir=constants.VOLUME_PATH_MODEL,
             allow_patterns=["*.json", "*.safetensors", "*.model"],
-            ignore_patterns=["*.msgpack", "*.h5", "*.bin"],
+            ignore_patterns=["*.msgpack", "*.h5", "*.bin", ".pt", ".pth"],
         )
 
         logging.info("Model has been downloaded")