Upodate model tag on loras

2026-04-20 14:29:34 +00:00 · 2025-04-19 10:41:27 -06:00
parent d455e76c4f
commit 77001ee77f
6 changed files with 35 additions and 12 deletions
--- a/toolkit/models/base_model.py
+++ b/toolkit/models/base_model.py
@@ -1486,3 +1486,7 @@ class BaseModel:
    def get_transformer_block_names(self) -> Optional[List[str]]:
        # override in child classes to get transformer block names for lora targeting
        return None
+    
+    def get_base_model_version() -> str:
+        # override in child classes to get the base model version
+        return "unknown"