Upodate model tag on loras

2026-03-13 06:29:48 +00:00 · 2025-04-19 10:41:27 -06:00
parent d455e76c4f
commit 77001ee77f
6 changed files with 35 additions and 12 deletions
--- a/extensions_built_in/diffusion_models/chroma/chroma_model.py
+++ b/extensions_built_in/diffusion_models/chroma/chroma_model.py
@@ -386,3 +386,6 @@ class ChromaModel(BaseModel):
            new_key = key.replace("diffusion_model.", "transformer.")
            new_sd[new_key] = value
        return new_sd
+    
+    def get_base_model_version(self):
+        return "chroma"
--- a/extensions_built_in/diffusion_models/hidream/hidream_model.py
+++ b/extensions_built_in/diffusion_models/hidream/hidream_model.py
@@ -443,3 +443,6 @@ class HidreamModel(BaseModel):
            new_sd[new_key] = value
        return new_sd
    
+    def get_base_model_version(self):
+        return "hidream_i1"
+