Skip to content

Problem with basicsr and torchvision.transforms.functional_tensor #1

@aurel-g

Description

@aurel-g

I installed all the requirements and have this error:

Traceback (most recent call last):
  File "/home/user/comfyui/nodes.py", line 2145, in load_custom_node
    module_spec.loader.exec_module(module)
  File "<frozen importlib._bootstrap_external>", line 940, in exec_module
  File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed
  File "/home/user/comfyui/custom_nodes/ComfyUI-RealESRGAN_Upscaler/__init__.py", line 4, in <module>
    from .realesrgan.realesrgan_upscaler import *
  File "/home/user/comfyui/custom_nodes/ComfyUI-RealESRGAN_Upscaler/realesrgan/realesrgan_upscaler.py", line 29, in <module>
    from basicsr.archs.rrdbnet_arch import RRDBNet
  File "/mnt/comfyui/venv/lib/python3.11/site-packages/basicsr/__init__.py", line 4, in <module>
    from .data import *
  File "/mnt/comfyui/venv/lib/python3.11/site-packages/basicsr/data/__init__.py", line 22, in <module>
    _dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames]
                       ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/mnt/comfyui/venv/lib/python3.11/site-packages/basicsr/data/__init__.py", line 22, in <listcomp>
    _dataset_modules = [importlib.import_module(f'basicsr.data.{file_name}') for file_name in dataset_filenames]
                        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/mnt/comfyui/venv/lib/python3.11/importlib/__init__.py", line 126, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/mnt/comfyui/venv/lib/python3.11/site-packages/basicsr/data/realesrgan_dataset.py", line 11, in <module>
    from basicsr.data.degradations import circular_lowpass_kernel, random_mixed_kernels
  File "/mnt/comfyui/venv/lib/python3.11/site-packages/basicsr/data/degradations.py", line 8, in <module>
    from torchvision.transforms.functional_tensor import rgb_to_grayscale

It may be related to this issue : XPixelGroup/BasicSR#533
Maybe there is a working combination of dependencies which works ?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions