diff --git a/deployment/transformers/utils/update_tiny_models.py b/deployment/transformers/utils/update_tiny_models.py
index 2e409906..f2eb9b71 100644
--- a/deployment/transformers/utils/update_tiny_models.py
+++ b/deployment/transformers/utils/update_tiny_models.py
@@ -29,7 +29,7 @@
 import time
 
 from create_dummy_models import COMPOSITE_MODELS, create_tiny_models
-from huggingface_hub import ModelFilter, hf_api
+from huggingface_hub import hf_api
 
 import transformers
 from transformers import AutoFeatureExtractor, AutoImageProcessor, AutoTokenizer
@@ -89,9 +89,7 @@ def get_tiny_model_summary_from_hub(output_path):
     # All tiny model base names on Hub
     model_names = get_all_model_names()
     models = hf_api.list_models(
-        filter=ModelFilter(
-            author="hf-internal-testing",
-        )
+        author="hf-internal-testing",
     )
     _models = set()
     for x in models:
diff --git a/gradients/utils/update_tiny_models.py b/gradients/utils/update_tiny_models.py
index 2e409906..f2eb9b71 100644
--- a/gradients/utils/update_tiny_models.py
+++ b/gradients/utils/update_tiny_models.py
@@ -29,7 +29,7 @@
 import time
 
 from create_dummy_models import COMPOSITE_MODELS, create_tiny_models
-from huggingface_hub import ModelFilter, hf_api
+from huggingface_hub import hf_api
 
 import transformers
 from transformers import AutoFeatureExtractor, AutoImageProcessor, AutoTokenizer
@@ -89,9 +89,7 @@ def get_tiny_model_summary_from_hub(output_path):
     # All tiny model base names on Hub
     model_names = get_all_model_names()
     models = hf_api.list_models(
-        filter=ModelFilter(
-            author="hf-internal-testing",
-        )
+        author="hf-internal-testing",
     )
     _models = set()
     for x in models:
diff --git a/quant/dbrx/utils/update_tiny_models.py b/quant/dbrx/utils/update_tiny_models.py
index 2e409906..f2eb9b71 100644
--- a/quant/dbrx/utils/update_tiny_models.py
+++ b/quant/dbrx/utils/update_tiny_models.py
@@ -29,7 +29,7 @@
 import time
 
 from create_dummy_models import COMPOSITE_MODELS, create_tiny_models
-from huggingface_hub import ModelFilter, hf_api
+from huggingface_hub import hf_api
 
 import transformers
 from transformers import AutoFeatureExtractor, AutoImageProcessor, AutoTokenizer
@@ -89,9 +89,7 @@ def get_tiny_model_summary_from_hub(output_path):
     # All tiny model base names on Hub
     model_names = get_all_model_names()
     models = hf_api.list_models(
-        filter=ModelFilter(
-            author="hf-internal-testing",
-        )
+        author="hf-internal-testing",
     )
     _models = set()
     for x in models: