Commit cb940a58 authored by AUTOMATIC's avatar AUTOMATIC

fix extension installation broken by #9518

parent 376e99f6
...@@ -129,7 +129,7 @@ def normalize_git_url(url): ...@@ -129,7 +129,7 @@ def normalize_git_url(url):
return url return url
def install_extension_from_url(dirname, branch_name, url): def install_extension_from_url(dirname, url, branch_name=None):
check_access() check_access()
assert url, 'No URL specified' assert url, 'No URL specified'
...@@ -150,7 +150,7 @@ def install_extension_from_url(dirname, branch_name, url): ...@@ -150,7 +150,7 @@ def install_extension_from_url(dirname, branch_name, url):
try: try:
shutil.rmtree(tmpdir, True) shutil.rmtree(tmpdir, True)
if branch_name == '': if not branch_name:
# if no branch is specified, use the default branch # if no branch is specified, use the default branch
with git.Repo.clone_from(url, tmpdir) as repo: with git.Repo.clone_from(url, tmpdir) as repo:
repo.remote().fetch() repo.remote().fetch()
...@@ -390,7 +390,7 @@ def create_ui(): ...@@ -390,7 +390,7 @@ def create_ui():
install_button.click( install_button.click(
fn=modules.ui.wrap_gradio_call(install_extension_from_url, extra_outputs=[gr.update()]), fn=modules.ui.wrap_gradio_call(install_extension_from_url, extra_outputs=[gr.update()]),
inputs=[install_dirname, install_branch, install_url], inputs=[install_dirname, install_url, install_branch],
outputs=[extensions_table, install_result], outputs=[extensions_table, install_result],
) )
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment