jenbenarye commited on
Commit
2762989
·
1 Parent(s): 3471c55

metadata loading and saving

Browse files
Files changed (1) hide show
  1. 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
- # Using a consistent naming pattern that links to the FEEL project
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