Skip to content

Commit

Permalink
Add import_hook hack to work around basicsr incompatibility
Browse files Browse the repository at this point in the history
Fixes #13985
  • Loading branch information
akx committed Dec 3, 2023
1 parent b4776ea commit d92ce14
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions modules/import_hook.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,3 +3,14 @@
# this will break any attempt to import xformers which will prevent stability diffusion repo from trying to use it
if "--xformers" not in "".join(sys.argv):
sys.modules["xformers"] = None

# Hack to fix a changed import in torchvision 0.17+, which otherwise breaks
# basicsr; see https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/13985
try:
import torchvision.transforms.functional_tensor # noqa: F401
except ImportError:
try:
import torchvision.transforms.functional as functional
sys.modules["torchvision.transforms.functional_tensor"] = functional
except ImportError:
pass # shrug...

0 comments on commit d92ce14

Please sign in to comment.