bug fix

intel · Nov 6, 2023 · 16be133 · 16be133
1 parent 68771fc
commit 16be133
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/e2eAIOK/deltatuner/deltatuner/deltatuner_model.py b/e2eAIOK/deltatuner/deltatuner/deltatuner_model.py
@@ -66,7 +66,7 @@ def __init__(self, model, peft_config: PeftConfig, adapter_name: str = "default"
         logging.basicConfig(level=logging.INFO,
                 format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
         self.logger = logging.getLogger('deltatuner')
-        if not isinstance(model, PeftModel):
+        if peft_config.peft_type in (DeltaTunerType.SSF):
             self.base_model = DELTATUNNER_TO_MODEL_MAPPING[peft_config.peft_type](
                 self.base_model, self.peft_config, adapter_name
             )

diff --git a/e2eAIOK/deltatuner/deltatuner/version b/e2eAIOK/deltatuner/deltatuner/version
@@ -1 +1 @@
-1.1.5
+1.1.6