Traceback (most recent call last):
File "/fsx/sayak/diffusers/check_kernelize.py", line 79, in <module>
main(args)
File "/fsx/sayak/diffusers/check_kernelize.py", line 57, in main
pipe = initialize_pipeline(args)
File "/fsx/sayak/diffusers/check_kernelize.py", line 27, in initialize_pipeline
pipe.transformer = kernelize(pipe.transformer, mode=Mode.INFERENCE | Mode.TORCH_COMPILE)
File "/fsx/sayak/kernels/src/kernels/layer.py", line 918, in kernelize
layer = _get_layer_memoize(repo, module_class)
File "/fsx/sayak/kernels/src/kernels/layer.py", line 1151, in _get_layer_memoize
_validate_layer(check_cls=module_class, cls=layer)
File "/fsx/sayak/kernels/src/kernels/layer.py", line 1027, in _validate_layer
raise TypeError(
TypeError: Forward signature does not match: different number of arguments.
Also include the class repo_id/layer name.