search

flowtyone / ComfyUI-Flowty-LDSR

LDSR custom node for ComfyUI
GNU General Public License v3.0
190 stars 14 forks source link

cannot import name 'measure_flops' from 'lightning_fabric.utilities' #15

Open curiosus0ne opened 6 months ago

curiosus0ne commented 6 months ago

Error occurred when executing LDSRUpscaler:

cannot import name 'measure_flops' from 'lightning_fabric.utilities' (D:\AIProjects\comfyui\env\lib\site-packages\lightning_fabric\utilities__init__.py)

File "D:\AIProjects\comfyui\execution.py", line 151, in recursive_execute output_data, output_ui = get_output_data(obj, input_data_all) File "D:\AIProjects\comfyui\execution.py", line 81, in get_output_data return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True) File "D:\AIProjects\comfyui\execution.py", line 74, in map_node_over_list results.append(getattr(obj, func)(*slice_dict(input_data_all, i))) File "D:\AIProjects\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 "D:\AIProjects\comfyui\env\lib\site-packages\torch\utils_contextlib.py", line 115, in decorate_context return func(args, **kwargs) File "D:\AIProjects\comfyui\custom_nodes\ComfyUI-Flowty-LDSR\ldsrlib\LDSR.py", line 261, in superResolution model = self.load_model_from_config() File "D:\AIProjects\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 "D:\AIProjects\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 "D:\AIProjects\comfyui\env\lib\site-packages\torch\serialization.py", line 1026, in load return _load(opened_zipfile, File "D:\AIProjects\comfyui\env\lib\site-packages\torch\serialization.py", line 1438, in _load result = unpickler.load() File "D:\AIProjects\comfyui\env\lib\site-packages\torch\serialization.py", line 1431, in find_class return super().find_class(mod_name, name) File "D:\AIProjects\comfyui\env\lib\site-packages\pytorch_lightning__init.py", line 27, in from pytorch_lightning.callbacks import Callback # noqa: E402 File "D:\AIProjects\comfyui\env\lib\site-packages\pytorch_lightning\callbacks__init__.py", line 14, in from pytorch_lightning.callbacks.batch_size_finder import BatchSizeFinder File "D:\AIProjects\comfyui\env\lib\site-packages\pytorch_lightning\callbacks\batch_size_finder.py", line 26, in from pytorch_lightning.callbacks.callback import Callback File "D:\AIProjects\comfyui\env\lib\site-packages\pytorch_lightning\callbacks\callback.py", line 22, in from pytorch_lightning.utilities.types import STEP_OUTPUT File "D:\AIProjects\comfyui\env\lib\site-packages\pytorch_lightning\utilities\init__.py", line 18, in from lightning_fabric.utilities import (

shiloh92 commented 6 months ago

cannot import name 'get_num_classes' from 'torchmetrics.utilities.data'