Open
Description
Your current environment information
Ran a simple workflow on ComfyUI.
🐛 Describe the bug
Getting below error
!!! Exception during processing !!! Transform failed of <class 'register_comfy.deep_cache_unet.FastDeepCacheUNet'>: Transform failed of <class 'onediff.infer_compiler.backends.oneflow.deployable_module.get_mixed_deployable_module.<locals>.MixedOneflowDeployableModule'>: An exception occurred during class transformation:
Traceback (most recent call last):
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 62, in proxy_class
out = transform_mgr.transform_cls(cls)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/manager.py", line 84, in transform_cls
mock_cls = self._transform_entity(mock_full_cls_name)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/manager.py", line 61, in _transform_entity
result = self.mocker.mock_entity(entity)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/import_tools/importer.py", line 95, in mock_entity
return self.load_entity_with_mock(entity)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/import_tools/importer.py", line 133, in load_entity_with_mock
mock_pkg = getattr(mock_pkg, name)
AttributeError: 'function' object has no attribute '<locals>'
Exception: 'function' object has no attribute '<locals>'
Traceback (most recent call last):
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 62, in proxy_class
out = transform_mgr.transform_cls(cls)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/manager.py", line 84, in transform_cls
mock_cls = self._transform_entity(mock_full_cls_name)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/manager.py", line 61, in _transform_entity
result = self.mocker.mock_entity(entity)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/import_tools/importer.py", line 95, in mock_entity
return self.load_entity_with_mock(entity)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/import_tools/importer.py", line 133, in load_entity_with_mock
mock_pkg = getattr(mock_pkg, name)
AttributeError: 'function' object has no attribute '<locals>'
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 41, in wrapper
result = dispatcher(first_param, *args, **kwargs)
File "/usr/lib/python3.10/functools.py", line 889, in wrapper
return dispatch(args[0].__class__)(*args, **kw)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 199, in _
new_md_cls = proxy_class(type(mod))
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 66, in proxy_class
raise RuntimeError(
RuntimeError: An exception occurred during class transformation:
Traceback (most recent call last):
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 62, in proxy_class
out = transform_mgr.transform_cls(cls)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/manager.py", line 84, in transform_cls
mock_cls = self._transform_entity(mock_full_cls_name)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/manager.py", line 61, in _transform_entity
result = self.mocker.mock_entity(entity)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/import_tools/importer.py", line 95, in mock_entity
return self.load_entity_with_mock(entity)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/import_tools/importer.py", line 133, in load_entity_with_mock
mock_pkg = getattr(mock_pkg, name)
AttributeError: 'function' object has no attribute '<locals>'
Exception: 'function' object has no attribute '<locals>'
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 41, in wrapper
result = dispatcher(first_param, *args, **kwargs)
File "/usr/lib/python3.10/functools.py", line 889, in wrapper
return dispatch(args[0].__class__)(*args, **kw)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 243, in _
of_mod = of_mod_cls()
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 214, in init
self._modules[n] = torch2oflow(m)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 43, in wrapper
raise NotImplementedError(f"Transform failed of {type(first_param)}: {e}")
NotImplementedError: Transform failed of <class 'onediff.infer_compiler.backends.oneflow.deployable_module.get_mixed_deployable_module.<locals>.MixedOneflowDeployableModule'>: An exception occurred during class transformation:
Traceback (most recent call last):
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 62, in proxy_class
out = transform_mgr.transform_cls(cls)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/manager.py", line 84, in transform_cls
mock_cls = self._transform_entity(mock_full_cls_name)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/manager.py", line 61, in _transform_entity
result = self.mocker.mock_entity(entity)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/import_tools/importer.py", line 95, in mock_entity
return self.load_entity_with_mock(entity)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/import_tools/importer.py", line 133, in load_entity_with_mock
mock_pkg = getattr(mock_pkg, name)
AttributeError: 'function' object has no attribute '<locals>'
Exception: 'function' object has no attribute '<locals>'
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/workspace/snap_comfy/ComfyUI/execution.py", line 327, in execute
output_data, output_ui, has_subgraph = get_output_data(obj, input_data_all, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb)
File "/workspace/snap_comfy/ComfyUI/execution.py", line 202, in get_output_data
return_values = _map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb)
File "/workspace/snap_comfy/ComfyUI/execution.py", line 174, in _map_node_over_list
process_inputs(input_dict, i)
File "/workspace/snap_comfy/ComfyUI/execution.py", line 163, in process_inputs
results.append(getattr(obj, func)(**inputs))
File "/workspace/snap_comfy/ComfyUI/nodes.py", line 1533, in sample
return common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive, negative, latent_image, denoise=denoise)
File "/workspace/snap_comfy/ComfyUI/nodes.py", line 1500, in common_ksampler
samples = comfy.sample.sample(model, noise, steps, cfg, sampler_name, scheduler, positive, negative, latent_image,
File "/workspace/snap_comfy/ComfyUI/custom_nodes/ComfyUI-Impact-Pack/modules/impact/sample_error_enhancer.py", line 22, in informative_sample
raise e
File "/workspace/snap_comfy/ComfyUI/custom_nodes/ComfyUI-Impact-Pack/modules/impact/sample_error_enhancer.py", line 9, in informative_sample
return original_sample(*args, **kwargs) # This code helps interpret error messages that occur within exceptions but does not have any impact on other operations.
File "/workspace/snap_comfy/ComfyUI/comfy/sample.py", line 45, in sample
samples = sampler.sample(noise, positive, negative, cfg=cfg, latent_image=latent_image, start_step=start_step, last_step=last_step, force_full_denoise=force_full_denoise, denoise_mask=noise_mask, sigmas=sigmas, callback=callback, disable_pbar=disable_pbar, seed=seed)
File "/workspace/snap_comfy/ComfyUI/comfy/samplers.py", line 1031, in sample
return sample(self.model, noise, positive, negative, cfg, self.device, sampler, sigmas, self.model_options, latent_image=latent_image, denoise_mask=denoise_mask, callback=callback, disable_pbar=disable_pbar, seed=seed)
File "/workspace/snap_comfy/ComfyUI/comfy/samplers.py", line 921, in sample
return cfg_guider.sample(noise, latent_image, sampler, sigmas, denoise_mask, callback, disable_pbar, seed)
File "/workspace/snap_comfy/ComfyUI/comfy/samplers.py", line 907, in sample
output = executor.execute(noise, latent_image, sampler, sigmas, denoise_mask, callback, disable_pbar, seed)
File "/workspace/snap_comfy/ComfyUI/comfy/patcher_extension.py", line 110, in execute
return self.original(*args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/comfy/samplers.py", line 876, in outer_sample
output = self.inner_sample(noise, latent_image, device, sampler, sigmas, denoise_mask, callback, disable_pbar, seed)
File "/workspace/snap_comfy/ComfyUI/comfy/samplers.py", line 860, in inner_sample
samples = executor.execute(self, sigmas, extra_args, callback, noise, latent_image, denoise_mask, disable_pbar)
File "/workspace/snap_comfy/ComfyUI/comfy/patcher_extension.py", line 110, in execute
return self.original(*args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/comfy/samplers.py", line 715, in sample
samples = self.sampler_function(model_k, noise, sigmas, extra_args=extra_args, callback=k_callback, disable=disable_pbar, **self.extra_options)
File "/workspace/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/comfy/k_diffusion/sampling.py", line 161, in sample_euler
denoised = model(x, sigma_hat * s_in, **extra_args)
File "/workspace/snap_comfy/ComfyUI/comfy/samplers.py", line 380, in __call__
out = self.inner_model(x, sigma, model_options=model_options, seed=seed)
File "/workspace/snap_comfy/ComfyUI/comfy/samplers.py", line 840, in __call__
return self.predict_noise(*args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/comfy/samplers.py", line 843, in predict_noise
return sampling_function(self.inner_model, x, timestep, self.conds.get("negative", None), self.conds.get("positive", None), self.cfg, model_options=model_options, seed=seed)
File "/workspace/snap_comfy/ComfyUI/comfy/samplers.py", line 360, in sampling_function
out = calc_cond_batch(model, conds, x, timestep, model_options)
File "/workspace/snap_comfy/ComfyUI/custom_nodes/onediff_comfy_nodes/modules/sd_hijack_utils.py", line 58, in hijacked_method
return self(*args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/custom_nodes/onediff_comfy_nodes/modules/sd_hijack_utils.py", line 95, in __call__
return sub_func(self._orig_func, *args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/custom_nodes/onediff_comfy_nodes/modules/oneflow/hijack_samplers.py", line 125, in calc_cond_batch_of
output = model_options["model_function_wrapper"](
File "/workspace/snap_comfy/ComfyUI/custom_nodes/onediff_comfy_nodes/modules/oneflow/utils/deep_cache_speedup.py", line 137, in apply_model
model_output, cache_h = model_patcher.fast_deep_cache_unet(
File "/workspace/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/workspace/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1582, in _call_impl
result = forward_call(*args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/deployable_module.py", line 38, in wrapper
return func(self, *args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/online_quantization_utils.py", line 65, in wrapper
output = func(self, *args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/graph_management_utils.py", line 123, in wrapper
ret = func(self, *args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/args_tree_util.py", line 72, in wrapper
graph_file_management(func)(self, *mapped_args, **mapped_kwargs)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/graph_management_utils.py", line 123, in wrapper
ret = func(self, *args, **kwargs)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/deployable_module.py", line 142, in forward
dpl_graph = self.get_graph()
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/deployable_module.py", line 108, in get_graph
self._deployable_module_model.oneflow_module,
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/dual_module.py", line 30, in oneflow_module
self._oneflow_module = torch2oflow(self._torch_module)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 43, in wrapper
raise NotImplementedError(f"Transform failed of {type(first_param)}: {e}")
NotImplementedError: Transform failed of <class 'register_comfy.deep_cache_unet.FastDeepCacheUNet'>: Transform failed of <class 'onediff.infer_compiler.backends.oneflow.deployable_module.get_mixed_deployable_module.<locals>.MixedOneflowDeployableModule'>: An exception occurred during class transformation:
Traceback (most recent call last):
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/builtin_transform.py", line 62, in proxy_class
out = transform_mgr.transform_cls(cls)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/manager.py", line 84, in transform_cls
mock_cls = self._transform_entity(mock_full_cls_name)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/transform/manager.py", line 61, in _transform_entity
result = self.mocker.mock_entity(entity)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/import_tools/importer.py", line 95, in mock_entity
return self.load_entity_with_mock(entity)
File "/workspace/snap_comfy/ComfyUI/onediff/src/onediff/infer_compiler/backends/oneflow/import_tools/importer.py", line 133, in load_entity_with_mock
mock_pkg = getattr(mock_pkg, name)
AttributeError: 'function' object has no attribute '<locals>'
Exception: 'function' object has no attribute '<locals>'