Spaces:
Running
on
A100
Running
on
A100
jenbenarye
commited on
Commit
·
2762989
1
Parent(s):
3471c55
metadata loading and saving
Browse files- ml/trainer.py +1 -10
ml/trainer.py
CHANGED
@@ -240,22 +240,13 @@ def main():
|
|
240 |
# Save metadata
|
241 |
metadata = AdapterMetadata(
|
242 |
training_timestamp=training_timestamp,
|
243 |
-
dataset_entries=[entry["id"] for entry in dataset],
|
244 |
-
training_params={
|
245 |
-
"max_weight": script_args.max_weight,
|
246 |
-
"min_weight": script_args.min_weight,
|
247 |
-
"decay_factor": script_args.decay_factor,
|
248 |
-
"training_mode": script_args.training_mode
|
249 |
-
},
|
250 |
model_name=model_args.model_name,
|
251 |
language=script_args.language,
|
252 |
-
version=training_timestamp
|
253 |
)
|
254 |
metadata.save(adapter_path / "metadata.json")
|
255 |
|
256 |
if script_args.push_to_hub:
|
257 |
-
|
258 |
-
repo_id = f"feel-fl/kto-lora-adapter-{script_args.language}"
|
259 |
print(f"Pushing adapter to Hugging Face Hub at {repo_id}...")
|
260 |
model.push_to_hub(repo_id=repo_id)
|
261 |
|
|
|
240 |
# Save metadata
|
241 |
metadata = AdapterMetadata(
|
242 |
training_timestamp=training_timestamp,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
243 |
model_name=model_args.model_name,
|
244 |
language=script_args.language,
|
|
|
245 |
)
|
246 |
metadata.save(adapter_path / "metadata.json")
|
247 |
|
248 |
if script_args.push_to_hub:
|
249 |
+
repo_id = f"feel-fl/adapters/{model_args.model_name.replace('/', '_')}/{script_args.language}"
|
|
|
250 |
print(f"Pushing adapter to Hugging Face Hub at {repo_id}...")
|
251 |
model.push_to_hub(repo_id=repo_id)
|
252 |
|