runtime error

Exit code: 1. Reason: odule> torch.manual_seed = disable(torch.manual_seed) File "/usr/local/lib/python3.10/site-packages/torch/_dynamo/decorators.py", line 50, in disable return DisableContext()(fn) File "/usr/local/lib/python3.10/site-packages/torch/_dynamo/eval_frame.py", line 410, in __call__ (filename is None or trace_rules.check(fn)) File "/usr/local/lib/python3.10/site-packages/torch/_dynamo/trace_rules.py", line 3378, in check return check_verbose(obj, is_inlined_call).skipped File "/usr/local/lib/python3.10/site-packages/torch/_dynamo/trace_rules.py", line 3361, in check_verbose rule = torch._dynamo.trace_rules.lookup_inner( File "/usr/local/lib/python3.10/site-packages/torch/_dynamo/trace_rules.py", line 3442, in lookup_inner rule = get_torch_obj_rule_map().get(obj, None) File "/usr/local/lib/python3.10/site-packages/torch/_dynamo/trace_rules.py", line 2782, in get_torch_obj_rule_map obj = load_object(k) File "/usr/local/lib/python3.10/site-packages/torch/_dynamo/trace_rules.py", line 2811, in load_object val = _load_obj_from_str(x[0]) File "/usr/local/lib/python3.10/site-packages/torch/_dynamo/trace_rules.py", line 2795, in _load_obj_from_str return getattr(importlib.import_module(module), obj_name) File "/usr/local/lib/python3.10/importlib/__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "/usr/local/lib/python3.10/site-packages/torch/nested/_internal/nested_tensor.py", line 419, in <module> ).detach() File "/usr/local/lib/python3.10/site-packages/spaces/zero/torch/patching.py", line 198, in __torch_function__ return transform_subclass(wrapper_tensor, lambda _, t: func(t)) File "/usr/local/lib/python3.10/site-packages/torch/utils/_python_dispatch.py", line 260, in transform_subclass assert sub.shape == outer_size, \ AssertionError: Expected return value from <class 'torch.nested._internal.nested_tensor.NestedTensor'>__tensor_unflatten__() to have shape equal to (1, j0, 3), but got: (1, j1, 3)

Container logs: