runtime error

Exit code: 1. Reason: ] loading stabilityai/stable-audio-3-medium … model_config.json: 0%| | 0.00/10.4k [00:00<?, ?B/s] model_config.json: 100%|██████████| 10.4k/10.4k [00:00<00:00, 33.9MB/s] No module named 'flash_attn' flash_attn not installed, disabling Flash Attention Traceback (most recent call last): File "/app/app.py", line 118, in <module> model, config = get_pretrained_model(v.repo) ~~~~~~~~~~~~~~~~~~~~^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/stable_audio_tools/models/pretrained.py", line 15, in get_pretrained_model model = create_model_from_config(model_config) File "/usr/local/lib/python3.13/site-packages/stable_audio_tools/models/factory.py", line 16, in create_model_from_config return create_diffusion_cond_from_config(model_config) File "/usr/local/lib/python3.13/site-packages/stable_audio_tools/models/diffusion.py", line 651, in create_diffusion_cond_from_config diffusion_model = DiTWrapper(diffusion_objective=diffusion_objective, **diffusion_model_config) File "/usr/local/lib/python3.13/site-packages/stable_audio_tools/models/diffusion.py", line 518, in __init__ self.model = DiffusionTransformer(diffusion_objective=diffusion_objective, *args, **kwargs) ~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/stable_audio_tools/models/dit.py", line 106, in __init__ self.transformer = ContinuousTransformer( ~~~~~~~~~~~~~~~~~~~~~^ dim=embed_dim, ^^^^^^^^^^^^^^ ...<7 lines>... **kwargs ^^^^^^^^ ) ^ File "/usr/local/lib/python3.13/site-packages/stable_audio_tools/models/transformer.py", line 782, in __init__ TransformerBlock( ~~~~~~~~~~~~~~~~^ dim, ^^^^ ...<8 lines>... **kwargs ^^^^^^^^ ) ^ TypeError: TransformerBlock.__init__() got an unexpected keyword argument 'local_add_cond_dim'. Did you mean 'global_cond_dim'?

Container logs: