SD3 Lora page filter - detection not implemented

This commit is contained in:
w-e-w
2024-07-31 02:43:13 +09:00
parent 48239090f1
commit 920a3a4dce
4 changed files with 24 additions and 4 deletions
+2
View File
@@ -19,6 +19,7 @@ class SdVersion(enum.Enum):
SD1 = 2
SD2 = 3
SDXL = 4
SD3 = 5
class NetworkOnDisk:
@@ -59,6 +60,7 @@ class NetworkOnDisk:
self.sd_version = self.detect_version()
def detect_version(self):
# TODO: SdVersion.SD3 detection
if str(self.metadata.get('ss_base_model_version', "")).startswith("sdxl_"):
return SdVersion.SDXL
elif str(self.metadata.get('ss_v2', "")) == "True":