diff --git a/clarifai/runners/utils/loader.py b/clarifai/runners/utils/loader.py index 2d3912eb..1fc63aef 100644 --- 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,