Update finetune_lora.py

finetune_lora.py

@@ -53,8 +53,8 @@ def train_lora(base_model: str, epochs: int = 2, lr: float = 1e-4, train_csv: st
     model = get_peft_model(model, lora_config)
     data_collator = DataCollatorForLanguageModeling(tokenizer=tokenizer, mlm=False)
 
-    #output_dir = "lora_adapter"
-    output_dir = os.path.join(os.getcwd(), "lora_adapter")
+    output_dir = "lora_adapter"
+    #output_dir = os.path.join(os.getcwd(), "lora_adapter")
     os.makedirs(output_dir, exist_ok=True)
 
     # Streamlit progress UI