diff --git a/e2eAIOK/deltatuner/deltatuner/deltatuner_model.py b/e2eAIOK/deltatuner/deltatuner/deltatuner_model.py index 0949d578c..596bfa9fb 100644 --- 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 index e25d8d9f3..0664a8fd2 100644 --- a/e2eAIOK/deltatuner/deltatuner/version +++ b/e2eAIOK/deltatuner/deltatuner/version @@ -1 +1 @@ -1.1.5 +1.1.6