Spaces:
Runtime error
Runtime error
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 |
|