Upodate model tag on loras

2026-04-26 09:19:20 +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/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"
+