torchvision 0.18.0+cu121节点导入失败

[vanch007]

Traceback (most recent call last):
File "D:\pinokio\api\comfyui\app\nodes.py", line 1864, in load_custom_node
module_spec.loader.exec_module(module)
File "", line 883, in exec_module
File "", line 241, in call_with_frames_removed
File "D:\pinokio\api\comfyui\app\custom_nodes\ComfyUI-PuLID-ZHO_init.py", line 1, in
from . import PuLID
File "D:\pinokio\api\comfyui\app\custom_nodes\ComfyUI-PuLID-ZHO\PuLID.py", line 9, in
from pulid.pipeline import PuLIDPipeline
File "D:\pinokio\api\comfyui\app\custom_nodes\ComfyUI-PuLID-ZHO\pulid\pipeline.py", line 7, in
from basicsr.utils import img2tensor, tensor2img
File "D:\pinokio\api\comfyui\app\env\lib\site-packages\basicsr_init_.py", line 4, in
from .data import *
File "D:\pinokio\api\comfyui\app\env\lib\site-packages\basicsr\data_init_.py", line 22, in
dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames]
File "D:\pinokio\api\comfyui\app\env\lib\site-packages\basicsr\data_init.py", line 22, in
dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames]
File "D:\pinokio\bin\miniconda\lib\importlib_init.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "D:\pinokio\api\comfyui\app\env\lib\site-packages\basicsr\data\realesrgan_dataset.py", line 11, in
from basicsr.data.degradations import circular_lowpass_kernel, random_mixed_kernels
File "D:\pinokio\api\comfyui\app\env\lib\site-packages\basicsr\data\degradations.py", line 8, in
from torchvision.transforms.functional_tensor import rgb_to_grayscale
ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'

Cannot import D:\pinokio\api\comfyui\app\custom_nodes\ComfyUI-PuLID-ZHO module for custom nodes: No module named 'torchvision.transforms.functional_tensor'

[kft334]

I modifed the degradations.py file in the basicsr package and now it works.

Change from:

from torchvision.transforms.functional_tensor import rgb_to_grayscale

To:

from torchvision.transforms import functional as tf

def rgb_to_grayscale(image):

    return tf.to_grayscale(image, num_output_channels=1)

[curlysasha]

degradations.py

same problem

[deepfree2023]

This maybe better.

Change:

from torchvision.transforms.functional_tensor import rgb_to_grayscale

To:

try: 
    from torchvision.transforms.functional_tensor import rgb_to_grayscale 
except: 
    from torchvision.transforms.functional import rgb_to_grayscale

[curlysasha]

not helped

This maybe better.

Change:

from torchvision.transforms.functional_tensor import rgb_to_grayscale

To:

try: 
    from torchvision.transforms.functional_tensor import rgb_to_grayscale 
except: 
    from torchvision.transforms.functional import rgb_to_grayscale