ImportError: cannot import name '_compare_version' from 'torchmetrics.utilities.imports'

[Ferocit]

Hey there,

wanted to give this a try, but I am getting an error in the "LDSR Upscale (all-in-one)" node:

Error occurred when executing LDSRUpscaler:

cannot import name '_compare_version' from 'torchmetrics.utilities.imports' (E:\Programme\ComfyUI_windows_portable\python_embeded\Lib\site-packages\torchmetrics\utilities\imports.py)

  File "E:\Programme\ComfyUI_windows_portable\ComfyUI\execution.py", line 152, in recursive_execute
    output_data, output_ui = get_output_data(obj, input_data_all)
                             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\ComfyUI\execution.py", line 82, in get_output_data
    return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True)
                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\ComfyUI\execution.py", line 75, in map_node_over_list
    results.append(getattr(obj, func)(**slice_dict(input_data_all, i)))
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-Flowty-LDSR\__init__.py", line 124, in upscale
    outputs.append(ldsr.superResolution(image, int(steps), pre_downscale, post_downscale, downsample_method))
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\python_embeded\Lib\site-packages\torch\utils\_contextlib.py", line 115, in decorate_context
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-Flowty-LDSR\ldsrlib\LDSR.py", line 261, in superResolution
    model = self.load_model_from_config()
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-Flowty-LDSR\ldsrlib\LDSR.py", line 88, in load_model_from_config
    self.model = LDSR.load_model_from_path(self.modelPath, self.torchdevice)
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-Flowty-LDSR\ldsrlib\LDSR.py", line 75, in load_model_from_path
    pl_sd = torch.load(modelPath, map_location="cpu")
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\python_embeded\Lib\site-packages\torch\serialization.py", line 1014, in load
    return _load(opened_zipfile,
           ^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\python_embeded\Lib\site-packages\torch\serialization.py", line 1422, in _load
    result = unpickler.load()
             ^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\python_embeded\Lib\site-packages\torch\serialization.py", line 1415, in find_class
    return super().find_class(mod_name, name)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\python_embeded\Lib\site-packages\pytorch_lightning\__init__.py", line 34, in 
    from pytorch_lightning.callbacks import Callback  # noqa: E402
    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\python_embeded\Lib\site-packages\pytorch_lightning\callbacks\__init__.py", line 25, in 
    from pytorch_lightning.callbacks.progress import ProgressBarBase, RichProgressBar, TQDMProgressBar
  File "E:\Programme\ComfyUI_windows_portable\python_embeded\Lib\site-packages\pytorch_lightning\callbacks\progress\__init__.py", line 22, in 
    from pytorch_lightning.callbacks.progress.rich_progress import RichProgressBar  # noqa: F401
    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\Programme\ComfyUI_windows_portable\python_embeded\Lib\site-packages\pytorch_lightning\callbacks\progress\rich_progress.py", line 20, in 
    from torchmetrics.utilities.imports import _compare_version

• Installed via ComfyUI Manager
• Model downloaded
• Torch version: 2.1.0+cu121

Settings:

• model: last.ckpt
• steps: 100
• pre_downscale: None
• post_downscale: None
• downsample_method: Lanczos