mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-10 06:17:55 -05:00
7c7e630099
* [SD] Add Scribble variant in Stencil -- This commit adds scribble variant in Stencil. Signed-off-by: Abhishek Varma <abhishek@nod-labs.com> * [SD] Use latest diffusers -- This commit points back to the latest diffusers and updates the processing script to tackle the Pix2Pix import issue. Signed-off-by: Abhishek Varma <abhishek@nod-labs.com> --------- Signed-off-by: Abhishek Varma <abhishek@nod-labs.com> Co-authored-by: Abhishek Varma <abhishek@nod-labs.com>
79 lines
2.9 KiB
Python
79 lines
2.9 KiB
Python
# This script will toggle the comment/uncommenting aspect for dealing
|
|
# with __file__ AttributeError arising in case of a few modules in
|
|
# `torch/_dynamo/skipfiles.py` (within shark.venv)
|
|
|
|
from distutils.sysconfig import get_python_lib
|
|
import fileinput
|
|
from pathlib import Path
|
|
|
|
# Diffusers 0.13.1 fails with transformers __init.py errros in BLIP. So remove it for now until we fork it
|
|
pix2pix_init = Path(get_python_lib() + "/diffusers/__init__.py")
|
|
for line in fileinput.input(pix2pix_init, inplace=True):
|
|
if "Pix2Pix" in line:
|
|
if not line.startswith("#"):
|
|
print(f"#{line}", end="")
|
|
else:
|
|
print(f"{line[1:]}", end="")
|
|
else:
|
|
print(line, end="")
|
|
pix2pix_init = Path(get_python_lib() + "/diffusers/pipelines/__init__.py")
|
|
for line in fileinput.input(pix2pix_init, inplace=True):
|
|
if "Pix2Pix" in line:
|
|
if not line.startswith("#"):
|
|
print(f"#{line}", end="")
|
|
else:
|
|
print(f"{line[1:]}", end="")
|
|
else:
|
|
print(line, end="")
|
|
pix2pix_init = Path(
|
|
get_python_lib() + "/diffusers/pipelines/stable_diffusion/__init__.py"
|
|
)
|
|
for line in fileinput.input(pix2pix_init, inplace=True):
|
|
if "StableDiffusionPix2PixZeroPipeline" in line:
|
|
if not line.startswith("#"):
|
|
print(f"#{line}", end="")
|
|
else:
|
|
print(f"{line[1:]}", end="")
|
|
else:
|
|
print(line, end="")
|
|
|
|
path_to_skipfiles = Path(get_python_lib() + "/torch/_dynamo/skipfiles.py")
|
|
|
|
modules_to_comment = ["abc,", "os,", "posixpath,", "_collections_abc,"]
|
|
startMonitoring = 0
|
|
for line in fileinput.input(path_to_skipfiles, inplace=True):
|
|
if "SKIP_DIRS = " in line:
|
|
startMonitoring = 1
|
|
print(line, end="")
|
|
elif startMonitoring in [1, 2]:
|
|
if "]" in line:
|
|
startMonitoring += 1
|
|
print(line, end="")
|
|
else:
|
|
flag = True
|
|
for module in modules_to_comment:
|
|
if module in line:
|
|
if not line.startswith("#"):
|
|
print(f"#{line}", end="")
|
|
else:
|
|
print(f"{line[1:]}", end="")
|
|
flag = False
|
|
break
|
|
if flag:
|
|
print(line, end="")
|
|
else:
|
|
print(line, end="")
|
|
|
|
# For getting around scikit-image's packaging, laze_loader has had a patch merged but yet to be released.
|
|
# Refer: https://github.com/scientific-python/lazy_loader
|
|
path_to_lazy_loader = Path(get_python_lib() + "/lazy_loader/__init__.py")
|
|
|
|
for line in fileinput.input(path_to_lazy_loader, inplace=True):
|
|
if 'stubfile = filename if filename.endswith("i")' in line:
|
|
print(
|
|
' stubfile = (filename if filename.endswith("i") else f"{os.path.splitext(filename)[0]}.pyi")',
|
|
end="",
|
|
)
|
|
else:
|
|
print(line, end="")
|