FIX: functional_tensor import error Linux/MacOS

Issue #161
Gourieff · Dec 24, 2023 · 58a8346 · 58a8346
1 parent efdd6af
commit 58a8346
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -2,7 +2,7 @@
 
   <img src="https://github.com/Gourieff/Assets/raw/main/sd-webui-reactor/ReActor_logo_NEW_EN.png?raw=true" alt="logo" width="180px"/>
 
-  ![Version](https://img.shields.io/badge/node_version-0.4.1_beta3-green?style=for-the-badge&labelColor=darkgreen)
+  ![Version](https://img.shields.io/badge/node_version-0.4.1_beta4-green?style=for-the-badge&labelColor=darkgreen)
 
   <sup>
   <font color=brightred>

diff --git a/README_RU.md b/README_RU.md
@@ -2,7 +2,7 @@
 
   <img src="https://github.com/Gourieff/Assets/raw/main/sd-webui-reactor/ReActor_logo_NEW_RU.png?raw=true" alt="logo" width="180px"/>
 
-  ![Version](https://img.shields.io/badge/версия_нода-0.4.1_beta3-green?style=for-the-badge&labelColor=darkgreen)
+  ![Version](https://img.shields.io/badge/версия_нода-0.4.1_beta4-green?style=for-the-badge&labelColor=darkgreen)
 
   <sup>
   <font color=brightred>

diff --git a/basicsr/data/degradations.py b/basicsr/data/degradations.py
@@ -5,7 +5,10 @@
 import torch
 from scipy import special
 from scipy.stats import multivariate_normal
-from torchvision.transforms.functional_tensor import rgb_to_grayscale
+try:
+    from torchvision.transforms.functional_tensor import rgb_to_grayscale
+except:
+    from torchvision.transforms.functional import rgb_to_grayscale
 
 # -------------------------------------------------------------------- #
 # --------------------------- blur kernels --------------------------- #

diff --git a/scripts/reactor_version.py b/scripts/reactor_version.py
@@ -1,5 +1,5 @@
 app_title = "ReActor Node for ComfyUI"
-version_flag = "v0.4.1-b3"
+version_flag = "v0.4.1-b4"
 
 from scripts.reactor_logger import logger