jenbenarye commited on
Commit
f7a79d4
·
1 Parent(s): 9635883

made adapter trainable + added timestamp to dirname

Browse files
Files changed (1) hide show
  1. ml/kto_lora.py +7 -4
ml/kto_lora.py CHANGED
@@ -126,7 +126,7 @@ def main():
126
 
127
  if os.path.isdir(adapter_dir):
128
  # If an adapter for this language already exists, load it into the base model.
129
- model = PeftModel.from_pretrained(model, adapter_dir)
130
  print(f"Loaded existing adapter for language '{script_args.language}' from {adapter_dir}.")
131
  else:
132
  # Otherwise, initialize a new LoRA adapter.
@@ -184,9 +184,12 @@ def main():
184
  # Adapter Saving
185
  # -----------------------------
186
  print("Saving adapter...")
187
- os.makedirs(adapter_dir, exist_ok=True)
188
- model.save_pretrained(adapter_dir)
189
- print(f"Adapter for language '{script_args.language}' saved to: {adapter_dir}")
 
 
 
190
 
191
  if script_args.push_to_hub:
192
  # Using a consistent naming pattern that links to the FEEL project
 
126
 
127
  if os.path.isdir(adapter_dir):
128
  # If an adapter for this language already exists, load it into the base model.
129
+ model = PeftModel.from_pretrained(model, adapter_dir, is_trainable=True)
130
  print(f"Loaded existing adapter for language '{script_args.language}' from {adapter_dir}.")
131
  else:
132
  # Otherwise, initialize a new LoRA adapter.
 
184
  # Adapter Saving
185
  # -----------------------------
186
  print("Saving adapter...")
187
+ # Add timestamp to adapter directory
188
+ timestamp = datetime.now().strftime('%Y-%m-%d_%H-%M-%S')
189
+ new_adapter_dir = os.path.join(adapter_dir, f"version_{timestamp}")
190
+ os.makedirs(new_adapter_dir, exist_ok=True)
191
+ model.save_pretrained(new_adapter_dir)
192
+ print(f"Adapter for language '{script_args.language}' saved to: {new_adapter_dir}")
193
 
194
  if script_args.push_to_hub:
195
  # Using a consistent naming pattern that links to the FEEL project