shibbir24 commited on
Commit
dacf175
·
verified ·
1 Parent(s): 2060be0

Update finetune_lora.py

Browse files
Files changed (1) hide show
  1. finetune_lora.py +2 -1
finetune_lora.py CHANGED
@@ -53,7 +53,8 @@ def train_lora(base_model: str, epochs: int = 2, lr: float = 1e-4, train_csv: st
53
  model = get_peft_model(model, lora_config)
54
  data_collator = DataCollatorForLanguageModeling(tokenizer=tokenizer, mlm=False)
55
 
56
- output_dir = "lora_adapter"
 
57
  os.makedirs(output_dir, exist_ok=True)
58
 
59
  # Streamlit progress UI
 
53
  model = get_peft_model(model, lora_config)
54
  data_collator = DataCollatorForLanguageModeling(tokenizer=tokenizer, mlm=False)
55
 
56
+ #output_dir = "lora_adapter"
57
+ output_dir = os.path.join(os.getcwd(), "lora_adapter")
58
  os.makedirs(output_dir, exist_ok=True)
59
 
60
  # Streamlit progress UI