From 71393c5801f8cbcbe47600db496530ab4c19d286 Mon Sep 17 00:00:00 2001
From: Maxime <91475935+maximedrn@users.noreply.github.com>
Date: Wed, 22 May 2024 21:25:45 +0200
Subject: [PATCH] fix: error when retrieving checkpoints in
 NovitaClient.models()

---
 src/novita_client/novita.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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(