Add base model mapping for SD 1.5

This commit is contained in:
Will Miao
2025-03-22 09:49:35 +08:00
parent 4cc6996406
commit 95380fbbfb

View File

@@ -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",