(self, args, kwargs)) File "/usr/local/lib/python3.10/dist-packages/torch/_dynamo/variables/nn_module.py", line 336, in call_function return tx.inline_user_function_return( File "/usr/local/lib/python3.10/dist-packages/torch/_dynamo/symbolic_convert.py", line 680, in inline_user_...