diff --git a/toolkit/models/base_model.py b/toolkit/models/base_model.py
index 0d5a821f..b371604f 100644
--- a/toolkit/models/base_model.py
+++ b/toolkit/models/base_model.py
@@ -1487,6 +1487,6 @@ class BaseModel:
         # override in child classes to get transformer block names for lora targeting
         return None
     
-    def get_base_model_version() -> str:
+    def get_base_model_version(self) -> str:
         # override in child classes to get the base model version
         return "unknown"