Spaces:
Running
on
Zero
Running
on
Zero
jenbenarye
commited on
Commit
·
f7a79d4
1
Parent(s):
9635883
made adapter trainable + added timestamp to dirname
Browse files- 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 |
-
|
188 |
-
|
189 |
-
|
|
|
|
|
|
|
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
|