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>
- 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)
|