From a3f8eb082a65aa65e3a34c623259b262e452da7f Mon Sep 17 00:00:00 2001
From: Jak Erdy <66637406+jakerdy@users.noreply.github.com>
Date: Mon, 12 May 2025 13:21:54 +0700
Subject: [PATCH] [Fix] Get Model Info local image url for non-lora models

If you make request to retrieve model info, that is not lora, first image link always will be created as if model is lora, but, it wouldn't work for checkpoints, unet, upscalers, etc.
---
 py/server/utils_info.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/py/server/utils_info.py b/py/server/utils_info.py
index e04b16fc..8c077535 100644
--- a/py/server/utils_info.py
+++ b/py/server/utils_info.py
@@ -81,7 +81,7 @@ async def get_model_info(file: str,
     should_save = True
 
   if img_next_to_file:
-    img_next_to_file_url = f'/rgthree/api/loras/img?file={file}'
+    img_next_to_file_url = f'/rgthree/api/{model_type}/img?file={file}'
     if len(info_data['images']) == 0 or info_data['images'][0]['url'] != img_next_to_file_url:
       info_data['images'].insert(0, {'url': img_next_to_file_url})
       should_save = True