diff --git a/src/novita_client/novita.py b/src/novita_client/novita.py
index ff6ba8d..1ca9f46 100644
--- a/src/novita_client/novita.py
+++ b/src/novita_client/novita.py
@@ -989,7 +989,7 @@ def models(self, refresh=False) -> ModelList:
         """
 
         if (self._model_list_cache is None or len(self._model_list_cache) == 0) or refresh:
-            res = self._get('/v2/models')
+            res = self._get('/v2/models', params={'type': 'checkpoint'})
 
             # TODO: fix this
             res_controlnet = self._get(