Add custom attributes to Node in JIT

Is there any way to add custom attributes to JIT Node ? I need to use this attribute when I run custom passes on JIT Graph.


Hi, did you find the solution?
I tried register_buffer and register_parameter APIs (they are on module level but not Node, but they fit my need), but neither works. I got below error when calling them for a loaded JIT model.
RuntimeError: Can’t add a new parameter after ScriptModule construction.
And below error when calling a new ScriptModule.
AttributeError: cannot assign buffer before Module.init() call
any input will be appreciated.