patrickvonplaten commited on
Commit
d89b237
1 Parent(s): 9ee08ed

Update convert.py

Browse files
Files changed (1) hide show
  1. convert.py +1 -1
convert.py CHANGED
@@ -197,7 +197,7 @@ def previous_pr(api: "HfApi", model_id: str, pr_title: str) -> Optional["Discuss
197
  def convert_generic(model_id: str, folder: str, filenames: Set[str]) -> List["CommitOperationAdd"]:
198
  operations = []
199
 
200
- extensions = set([".bin", ".ckpt"])
201
  for filename in filenames:
202
  prefix, ext = os.path.splitext(filename)
203
  if ext in extensions:
 
197
  def convert_generic(model_id: str, folder: str, filenames: Set[str]) -> List["CommitOperationAdd"]:
198
  operations = []
199
 
200
+ extensions = set([".bin"])
201
  for filename in filenames:
202
  prefix, ext = os.path.splitext(filename)
203
  if ext in extensions: