diff --git a/python/freetoken/utils/hf.py b/python/freetoken/utils/hf.py index 5a6a31f2..ee403be2 100644 --- a/python/freetoken/utils/hf.py +++ b/python/freetoken/utils/hf.py @@ -210,7 +210,7 @@ def download_hf_weight(model_path: str) -> str: try: return snapshot_download( model_path, - allow_patterns=["*.safetensors"], + allow_patterns=["*.safetensors", "*.json"], tqdm_class=DisabledTqdm, ) except Exception as e: