update

asset3d_gen/models/sr_model.py

@@ -58,6 +58,17 @@ class ImageStableSR:
 
 class ImageRealESRGAN:
     def __init__(self, outscale: int, model_path: str = None) -> None:
+        # monkey_patch_basicsr.py
+        import sys
+        import types
+        import torchvision
+        from packaging import version
+        if version.parse(torchvision.__version__) >= version.parse("0.16"):
+            import torchvision.transforms.functional as TF
+            functional_tensor = types.ModuleType("torchvision.transforms.functional_tensor")
+            functional_tensor.rgb_to_grayscale = TF.rgb_to_grayscale
+            sys.modules["torchvision.transforms.functional_tensor"] = functional_tensor
+
         from basicsr.archs.rrdbnet_arch import RRDBNet
         from realesrgan import RealESRGANer