import_hook.py 654 B

12345678910111213141516
  1. import sys
  2. # this will break any attempt to import xformers which will prevent stability diffusion repo from trying to use it
  3. if "--xformers" not in "".join(sys.argv):
  4. sys.modules["xformers"] = None
  5. # Hack to fix a changed import in torchvision 0.17+, which otherwise breaks
  6. # basicsr; see https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/13985
  7. try:
  8. import torchvision.transforms.functional_tensor # noqa: F401
  9. except ImportError:
  10. try:
  11. import torchvision.transforms.functional as functional
  12. sys.modules["torchvision.transforms.functional_tensor"] = functional
  13. except ImportError:
  14. pass # shrug...