From 12e3095d8a423c4a89b5dd87e8229d681f07695e Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 19 Apr 2025 14:34:01 -0600 Subject: [PATCH] Fixed issue with saving base model version --- toolkit/models/base_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"