Fix for working with SDNext
parent
26450143c5
commit
55cbff67c0
|
|
@ -16,15 +16,22 @@ script_path = os.path.realpath(__file__)
|
|||
requirements_path = os.path.join(sys.path[2], "requirements.txt")
|
||||
config_path = os.path.join(sys.path[2], "config.json")
|
||||
|
||||
# Define vars
|
||||
gradio_version = None
|
||||
should_script_work = False
|
||||
|
||||
# Read the version of 'gradio' from 'requirements.txt'
|
||||
if os.path.isfile(requirements_path):
|
||||
with open(requirements_path, "r") as file:
|
||||
lines = file.readlines()
|
||||
gradio_line = next((line for line in lines if line.startswith("gradio")), None)
|
||||
gradio_version = gradio_line.split("==")[1].strip() if gradio_line else None
|
||||
should_script_work = True
|
||||
|
||||
# Check if config.json exists
|
||||
if os.path.isfile(config_path):
|
||||
# In case we have not found anything, we skip this script.
|
||||
if should_script_work:
|
||||
# Check if config.json exists
|
||||
if os.path.isfile(config_path):
|
||||
with open(config_path, 'r') as config_file:
|
||||
config = json.load(config_file)
|
||||
# Check if "canvas-zoom" is in the list of disabled_extensions
|
||||
|
|
@ -37,10 +44,9 @@ if os.path.isfile(config_path):
|
|||
launch.run(f'"{python}" -m pip install --force-reinstall --no-deps gradio=={gradio_version}', desc=f"Uninstalling modified gradio for canvas-zoom", errdesc=f"Couldn't uninstall canvas-zoom", live=False)
|
||||
os.remove(script_path)
|
||||
|
||||
# Check if the folder exists
|
||||
if not os.path.exists(canvasZoomPath) and gradio_version is not None:
|
||||
# Check if the folder exists
|
||||
if not os.path.exists(canvasZoomPath) and gradio_version is not None:
|
||||
# If the folder does not exist and we found 'gradio' version, uninstall 'gradio' and install it again
|
||||
launch.run(f'"{python}" -m pip install --force-reinstall --no-deps gradio=={gradio_version}', desc=f"Uninstalling modified gradio for canvas-zoom", errdesc=f"Couldn't uninstall canvas-zoom", live=False)
|
||||
|
||||
# Deleting the file
|
||||
os.remove(script_path)
|
||||
Loading…
Reference in New Issue