flowtyone / ComfyUI-Flowty-LDSR

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

Error trying to run the comfyui node #19

Open bejaranoo opened 3 months ago

bejaranoo commented 3 months ago

running on the latest version of comfyui

Loading model from C:\Users\LBM\Documents\StabilityMatrix-win-x64\Data\models\ESRGAN\4xNMKDSuperscale_4xNMKDSuperscale.pt
!!! Exception during processing!!! 'state_dict'
Traceback (most recent call last):
  File "C:\Users\LBM\ComfyUI\execution.py", line 151, in recursive_execute
    output_data, output_ui = get_output_data(obj, input_data_all)
  File "C:\Users\LBM\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 "C:\Users\LBM\ComfyUI\execution.py", line 74, in map_node_over_list
    results.append(getattr(obj, func)(**slice_dict(input_data_all, i)))
  File "C:\Users\LBM\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 "C:\Users\LBM\anaconda3\envs\comfyui\lib\site-packages\torch\utils\_contextlib.py", line 115, in decorate_context
    return func(*args, **kwargs)
  File "C:\Users\LBM\ComfyUI\custom_nodes\ComfyUI-Flowty-LDSR\ldsrlib\LDSR.py", line 261, in superResolution
    model = self.load_model_from_config()
  File "C:\Users\LBM\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 "C:\Users\LBM\ComfyUI\custom_nodes\ComfyUI-Flowty-LDSR\ldsrlib\LDSR.py", line 76, in load_model_from_path
    sd = pl_sd["state_dict"]
KeyError: 'state_dict'

Prompt executed in 0.07 seconds

image