Description
Your current environment information
OneDiff version: 1.2.1.dev6
🐛 Describe the bug
File "/workspace/lib/BrushNet/src/diffusers/pipelines/brushnet/pipeline_brushnet_sd_xl.py", line 1453, in call
noise_pred = self.unet(
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 1568, in _call_impl
result = forward_call(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/onediff/infer_compiler/backends/oneflow/deployable_module.py", line 38, in wrapper
return func(self, *args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/onediff/infer_compiler/backends/oneflow/online_quantization_utils.py", line 65, in wrapper
output = func(self, *args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/onediff/infer_compiler/backends/oneflow/graph_management_utils.py", line 123, in wrapper
ret = func(self, *args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/onediff/infer_compiler/backends/oneflow/args_tree_util.py", line 72, in wrapper
graph_file_management(func)(self, *mapped_args, **mapped_kwargs)
File "/usr/local/lib/python3.8/dist-packages/onediff/infer_compiler/backends/oneflow/graph_management_utils.py", line 123, in wrapper
ret = func(self, *args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/onediff/infer_compiler/backends/oneflow/deployable_module.py", line 144, in forward
output = dpl_graph(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/oneflow/nn/graph/graph.py", line 295, in call
self._compile(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/oneflow/nn/graph/graph.py", line 861, in _compile
return self._dynamic_input_graph_cache._compile(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/oneflow/nn/graph/cache.py", line 121, in _compile
return graph._compile(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/oneflow/nn/graph/graph.py", line 865, in _compile
return self._compile_new(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/oneflow/nn/graph/graph.py", line 884, in _compile_new
_, eager_outputs = self.build_graph(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/oneflow/nn/graph/graph.py", line 1429, in build_graph
outputs = self.__build_graph(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/oneflow/nn/graph/graph.py", line 1577, in __build_graph
outputs = self.build(*lazy_args, **lazy_kwargs)
File "/usr/local/lib/python3.8/dist-packages/onediff/infer_compiler/backends/oneflow/graph.py", line 19, in build
return self.model(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/oneflow/nn/graph/proxy.py", line 188, in call
result = self.__block_forward(*args, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/oneflow/nn/graph/proxy.py", line 239, in __block_forward
result = unbound_forward_of_module_instance(self, *args, **kwargs)
TypeError: forward() got an unexpected keyword argument 'down_block_add_samples'