From 9441fbd5a2f99bc1a8cfd3aa17d5fd810c3791d9 Mon Sep 17 00:00:00 2001
From: techjarves <usaramwasi99@Gmail.com>
Date: Tue, 23 Jun 2026 08:10:28 +0530
Subject: [PATCH] feat: create ModelManager component for managing image, text,
 speech, and TTS models

---
 app/frontend/src/components/ModelManager.jsx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/app/frontend/src/components/ModelManager.jsx b/app/frontend/src/components/ModelManager.jsx
index 81752c2..8a0c6d2 100644
--- a/app/frontend/src/components/ModelManager.jsx
+++ b/app/frontend/src/components/ModelManager.jsx
@@ -1103,7 +1103,7 @@ function ModelManager({
   };
 
   const handleUnloadModel = async (runtimeOverride = null) => {
-    const runtime = runtimeOverride || getRuntimeForType(activeModelType);
+    const runtime = runtimeOverride?.type ? runtimeOverride : getRuntimeForType(activeModelType);
     if (!runtime || isUnloading) return;
 
     if (runtime.type === "image") {
@@ -1541,7 +1541,7 @@ function ModelManager({
                   
                   <div style={{ display: "flex", gap: "8px" }}>
                     {isActive ? (
-                      <button className="m3-btn m3-btn-error" style={{ height: "36px", padding: "0 16px" }} onClick={handleUnloadModel} disabled={isUnloading}>
+                      <button className="m3-btn m3-btn-error" style={{ height: "36px", padding: "0 16px" }} onClick={() => handleUnloadModel()} disabled={isUnloading}>
                         {isUnloading ? <RefreshCw className="progress-spinner" size={14} /> : <Trash2 size={14} />}
                         <span>{isUnloading ? "Unloading" : "Unload"}</span>
                       </button>