Update convert.py
Browse files- convert.py +2 -2
convert.py
CHANGED
@@ -20,7 +20,7 @@ COMMIT_DESCRIPTION = """
|
|
20 |
This is an automated PR create with https://huggingface.co/spaces/safetensors/convert
|
21 |
|
22 |
This new file is equivalent to `pytorch_model.bin` but safe in the sense that
|
23 |
-
no arbitrary code can be put into it.
|
24 |
|
25 |
These files also happen to load much faster than their pytorch counterpart:
|
26 |
https://colab.research.google.com/github/huggingface/notebooks/blob/main/safetensors_doc/en/speed.ipynb
|
@@ -269,7 +269,7 @@ def convert(api: "HfApi", model_id: str, force: bool = False) -> Optional["Commi
|
|
269 |
repo_id=model_id,
|
270 |
operations=operations,
|
271 |
commit_message=pr_title,
|
272 |
-
commit_description=COMMIT_DESCRIPTION
|
273 |
create_pr=True,
|
274 |
)
|
275 |
print(f"Pr created at {new_pr.pr_url}")
|
|
|
20 |
This is an automated PR create with https://huggingface.co/spaces/safetensors/convert
|
21 |
|
22 |
This new file is equivalent to `pytorch_model.bin` but safe in the sense that
|
23 |
+
no arbitrary code can be put into it.
|
24 |
|
25 |
These files also happen to load much faster than their pytorch counterpart:
|
26 |
https://colab.research.google.com/github/huggingface/notebooks/blob/main/safetensors_doc/en/speed.ipynb
|
|
|
269 |
repo_id=model_id,
|
270 |
operations=operations,
|
271 |
commit_message=pr_title,
|
272 |
+
commit_description=COMMIT_DESCRIPTION,
|
273 |
create_pr=True,
|
274 |
)
|
275 |
print(f"Pr created at {new_pr.pr_url}")
|