Narsil HF staff olivierdehaene HF staff commited on
Commit
650d083
1 Parent(s): 849eff4

fix: Fix convert_multi method (#3)

Browse files

- fix: Fix convert_multi method (f6b3632cdf52adf6a78cd2204efb414fc293fd22)


Co-authored-by: Olivier Dehaene <olivierdehaene@users.noreply.huggingface.co>

Files changed (1) hide show
  1. convert.py +1 -1
convert.py CHANGED
@@ -47,7 +47,7 @@ def rename(pt_filename: str) -> str:
47
  return local
48
 
49
 
50
- def convert_multi(model_id: str) -> List["CommitOperationAdd"]:
51
  filename = hf_hub_download(repo_id=model_id, filename="pytorch_model.bin.index.json")
52
  with open(filename, "r") as f:
53
  data = json.load(f)
 
47
  return local
48
 
49
 
50
+ def convert_multi(model_id: str, folder: str) -> List["CommitOperationAdd"]:
51
  filename = hf_hub_download(repo_id=model_id, filename="pytorch_model.bin.index.json")
52
  with open(filename, "r") as f:
53
  data = json.load(f)