Discord bot: fix model caching (part 1/999)

VoltaML · Jun 19, 2024 · 4f444c5 · 4f444c5
1 parent 2e63c29
commit 4f444c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bot/models.py b/bot/models.py
@@ -101,7 +101,7 @@ async def load_model(self, ctx: Context, model: str) -> None:
 
         if status == 200:
             await message.edit(content=f"{response['message']}: {model}")
-            await shared_bot.models.refresh_available_models()
+            await shared_bot.models.refresh_loaded_models()
         else:
             await message.edit(content=f"Error: **{response.get('detail')}**")
 
@@ -122,7 +122,7 @@ async def unload_model(self, ctx: Context, model: str) -> None:
 
         if status == 200:
             await message.edit(content=f"{response['message']}: {model}")
-            await shared_bot.models.refresh_available_models()
+            await shared_bot.models.refresh_loaded_models()
         else:
             await message.edit(content=f"Error: {status}")