From 95380fbbfb86d1a146867e5ee775dbceb050f1eb Mon Sep 17 00:00:00 2001
From: Will Miao <13051207myq@gmail.com>
Date: Sat, 22 Mar 2025 09:49:35 +0800
Subject: [PATCH] Add base model mapping for SD 1.5

---
 py/utils/model_utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/py/utils/model_utils.py b/py/utils/model_utils.py
index 7f62925d..a5675d03 100644
--- a/py/utils/model_utils.py
+++ b/py/utils/model_utils.py
@@ -2,6 +2,7 @@ from typing import Optional
 
 # Base model mapping based on version string
 BASE_MODEL_MAPPING = {
+    "sd_1.5": "SD 1.5",
     "sd-v1-5": "SD 1.5",
     "sd-v2-1": "SD 2.1",
     "sdxl": "SDXL 1.0",