mrcuddle commited on
Commit
ea54467
·
verified ·
1 Parent(s): dd47c8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -61,9 +61,9 @@ def merge_and_upload(base_model, model_to_merge, scaling_factor, weight_drop_pro
61
  api.create_repo(repo_id=repo_name, token=token, exist_ok=True)
62
 
63
  # Upload the file
64
- api.upload_file(
65
- path_or_fileobj=outpath,
66
- path_in_repo="merged_model.safetensors",
67
  repo_id=repo_name,
68
  token=token
69
  )
 
61
  api.create_repo(repo_id=repo_name, token=token, exist_ok=True)
62
 
63
  # Upload the file
64
+ api.upload_folder(
65
+ folder_path=outpath,
66
+ repo_type=model,
67
  repo_id=repo_name,
68
  token=token
69
  )