although i'm not sure how it happened, this prevents #1

LSX-UniWue · Jul 1, 2024 · d942cdf · d942cdf
1 parent 40ab7ff
commit d942cdf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils.py b/src/utils.py
@@ -45,8 +45,8 @@ def get_max_seq_length(cfg: DictConfig) -> int:
 def get_LORA_model(model, cfg: DictConfig):
     from transformers.models.auto.configuration_auto import AutoConfig
 
+    model_conf = AutoConfig.from_pretrained(cfg["model"]["model_name"])
     if peft_config.task_type == "???":
-        model_conf = AutoConfig.from_pretrained(cfg["model"]["model_name"])
         if cfg["task"]["framework"] in ["sentence_transformer", "flair"]:
             peft_config.task_type = TaskType.FEATURE_EXTRACTION
         elif cfg["task"]["framework"] == "hf_qa":