Narsil HF staff commited on
Commit
7a97ac2
1 Parent(s): 0d4f218

Update convert.py

Browse files
Files changed (1) hide show
  1. convert.py +17 -0
convert.py CHANGED
@@ -16,6 +16,22 @@ from transformers import AutoConfig
16
  from transformers.pipelines.base import infer_framework_load_model
17
 
18
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
  class AlreadyExists(Exception):
20
  pass
21
 
@@ -253,6 +269,7 @@ def convert(api: "HfApi", model_id: str, force: bool = False) -> Optional["Commi
253
  repo_id=model_id,
254
  operations=operations,
255
  commit_message=pr_title,
 
256
  create_pr=True,
257
  )
258
  print(f"Pr created at {new_pr.pr_url}")
 
16
  from transformers.pipelines.base import infer_framework_load_model
17
 
18
 
19
+ 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
27
+
28
+ The widgets on your model page will run using this model even if this is not merged
29
+ making sure the file actually works.
30
+
31
+ Feel free to ignore this PR.
32
+ """
33
+
34
+
35
  class AlreadyExists(Exception):
36
  pass
37
 
 
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}")