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):
return url
def install_extension_from_url(dirname, branch_name, url):
def install_extension_from_url(dirname, url, branch_name=None):
check_access()
assert url, 'No URL specified'
......@@ -150,7 +150,7 @@ def install_extension_from_url(dirname, branch_name, url):
try:
shutil.rmtree(tmpdir, True)
if branch_name == '':
if not branch_name:
# if no branch is specified, use the default branch
with git.Repo.clone_from(url, tmpdir) as repo:
repo.remote().fetch()
......@@ -390,7 +390,7 @@ def create_ui():
install_button.click(
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],
)
......
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