Fix basicsr import issues with newer torchvision versions

Author: cmdr2

sdkit/models/model_loader/gfpgan.py

@@ -1,4 +1,21 @@
 import torch
+
+# hack for basicsr https://github.com/XPixelGroup/BasicSR/pull/650
+# credit: https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/14186/files
+
+import sys
+
+try:
+    import torchvision.transforms.functional_tensor
+except ImportError:
+    try:
+        import torchvision.transforms.functional as functional
+
+        sys.modules["torchvision.transforms.functional_tensor"] = functional
+    except ImportError:
+        pass
+# /hack
+
 from gfpgan import GFPGANer
 
 from sdkit import Context

sdkit/models/model_loader/realesrgan.py

@@ -1,6 +1,23 @@
 import os
 
 import torch
+
+# hack for basicsr https://github.com/XPixelGroup/BasicSR/pull/650
+# credit: https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/14186/files
+
+import sys
+
+try:
+    import torchvision.transforms.functional_tensor
+except ImportError:
+    try:
+        import torchvision.transforms.functional as functional
+
+        sys.modules["torchvision.transforms.functional_tensor"] = functional
+    except ImportError:
+        pass
+# /hack
+
 from basicsr.archs.rrdbnet_arch import RRDBNet
 from realesrgan import RealESRGANer