Update sd-scripts

bmaltais · Nov 2, 2024 · 18e587d · 18e587d
1 parent fe9066d
commit 18e587d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/kohya_gui/sd_modeltype.py b/kohya_gui/sd_modeltype.py
@@ -1,9 +1,12 @@
 from os.path import isfile
 from safetensors import safe_open
 import enum
+from .custom_logging import setup_logging
 
-# methodology is based on https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/82a973c04367123ae98bd9abdf80d9eda9b910e2/modules/sd_models.py#L379-L403
+# Set up logging
+log = setup_logging()
 
+# methodology is based on https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/82a973c04367123ae98bd9abdf80d9eda9b910e2/modules/sd_models.py#L379-L403
 
 class ModelType(enum.Enum):
     UNKNOWN = 0
@@ -44,6 +47,9 @@ def hasKeyPrefix(pfx):
                 self.model_type = ModelType.SD2
             elif hasKeyPrefix("model."):
                 self.model_type = ModelType.SD1
+            else:
+                log.warning("Unknown model type")
+                log.warning(st.keys())
         except:
             pass
 

diff --git a/sd-scripts b/sd-scripts