extend ignore pattern to subfolders (#485)

Clarifai · Jan 15, 2025 · 7126975 · 7126975
1 parent bc44143
commit 7126975
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clarifai/runners/utils/loader.py b/clarifai/runners/utils/loader.py
@@ -57,7 +57,7 @@ def download_checkpoints(self, checkpoint_path: str):
         repo_files = list_repo_files(repo_id=self.repo_id, token=self.token)
         if any(f.endswith(".safetensors") for f in repo_files):
           logger.info(f"SafeTensors found in {self.repo_id}, downloading only .safetensors files.")
-          ignore_patterns = ["original/*", "*.pth", "*.bin"]
+          ignore_patterns = ["**/original/*", "**/*.pth", "**/*.bin"]
         snapshot_download(
             repo_id=self.repo_id,
             local_dir=checkpoint_path,