Update scripts_infer/anidoc_inference.py

scripts_infer/anidoc_inference.py

@@ -1,18 +1,23 @@
 import sys
 import os
+import types
 
 from pyparsing import col
 sys.path.insert(0, ".")
 
-# --- compatibility shim for old diffusers on modern huggingface_hub ---
+# ---- huggingface_hub compatibility for old diffusers ----
 import huggingface_hub
-
 if not hasattr(huggingface_hub, "cached_download"):
-    # For this project, the symbol is mainly needed so old diffusers can import.
-    # In AniDoc's normal path, hf_hub_download is the closest modern equivalent.
     from huggingface_hub import hf_hub_download
     huggingface_hub.cached_download = hf_hub_download
-# ----------------------------------------------------------------------
+
+# ---- torchvision compatibility for old basicsr ----
+import torchvision.transforms.functional as TVF
+if "torchvision.transforms.functional_tensor" not in sys.modules:
+    functional_tensor = types.ModuleType("torchvision.transforms.functional_tensor")
+    functional_tensor.rgb_to_grayscale = TVF.rgb_to_grayscale
+    sys.modules["torchvision.transforms.functional_tensor"] = functional_tensor
+# ---------------------------------------------------------
 
 import argparse
 from packaging import version