Uncomment PR creation
Browse filescc
@narsil
- convert.py +6 -6
convert.py
CHANGED
@@ -183,12 +183,12 @@ def convert(api: "HfApi", model_id: str, force: bool=False) -> Optional["CommitI
|
|
183 |
|
184 |
if operations:
|
185 |
check_final_model(model_id, folder)
|
186 |
-
|
187 |
-
|
188 |
-
|
189 |
-
|
190 |
-
|
191 |
-
|
192 |
finally:
|
193 |
shutil.rmtree(folder)
|
194 |
return new_pr
|
|
|
183 |
|
184 |
if operations:
|
185 |
check_final_model(model_id, folder)
|
186 |
+
new_pr = api.create_commit(
|
187 |
+
repo_id=model_id,
|
188 |
+
operations=operations,
|
189 |
+
commit_message=pr_title,
|
190 |
+
create_pr=True,
|
191 |
+
)
|
192 |
finally:
|
193 |
shutil.rmtree(folder)
|
194 |
return new_pr
|