Spaces:
Running
Running
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)
|