TransformerBlock.forward() got unexpected keyword 'seconds_total'
Please help.. Getting this error when running inference.. not sure why this is coming up when I just took the example form the model card..
getting this same error on python 3.8 as well would appreciate any help can't figure out why I'm getting this error :(
Verified that torch.cuda.is_available() showing up as true and can see the model loaded in VRAM (A10G)
Full error: Traceback (most recent call last): ... File "/opt/conda/envs/audio/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl return forward_call(*args, **kwargs) File "/opt/conda/envs/audio/lib/python3.10/site-packages/stable_audio_tools/models/transformer.py", line 795, in forward x = checkpoint(layer, x, rotary_pos_emb = rotary_pos_emb, global_cond=global_cond, **kwargs) File "/opt/conda/envs/audio/lib/python3.10/site-packages/stable_audio_tools/models/transformer.py", line 27, in checkpoint return torch.utils.checkpoint.checkpoint(function, *args, **kwargs) File "/opt/conda/envs/audio/lib/python3.10/site-packages/torch/_compile.py", line 24, in inner return torch._dynamo.disable(fn, recursive)(*args, **kwargs) File "/opt/conda/envs/audio/lib/python3.10/site-packages/torch/_dynamo/eval_frame.py", line 451, in _fn return fn(*args, **kwargs) File "/opt/conda/envs/audio/lib/python3.10/site-packages/torch/_dynamo/external_utils.py", line 36, in inner return fn(*args, **kwargs) File "/opt/conda/envs/audio/lib/python3.10/site-packages/torch/utils/checkpoint.py", line 494, in checkpoint ret = function(*args, **kwargs) File "/opt/conda/envs/audio/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/opt/conda/envs/audio/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl return forward_call(*args, **kwargs) TypeError: TransformerBlock.forward() got an unexpected keyword argument 'seconds_total'