runtime error

Exit code: 1. Reason: peline = TrellisImageTo3DPipeline.from_pretrained("Stable-X/trellis-normal-v0-1") File "/home/user/app/trellis/pipelines/trellis_image_to_3d.py", line 56, in from_pretrained pipeline = super(TrellisImageTo3DPipeline, TrellisImageTo3DPipeline).from_pretrained(path) File "/home/user/app/trellis/pipelines/base.py", line 39, in from_pretrained _models = { File "/home/user/app/trellis/pipelines/base.py", line 40, in <dictcomp> k: models.from_pretrained(f"{path}/{v}") File "/home/user/app/trellis/models/__init__.py", line 59, in from_pretrained model = __getattr__(config['name'])(**config['args'], **kwargs) File "/home/user/app/trellis/models/__init__.py", line 22, in __getattr__ module = importlib.import_module(f".{module_name}", __name__) File "/usr/local/lib/python3.10/importlib/__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "<frozen importlib._bootstrap>", line 1050, in _gcd_import File "<frozen importlib._bootstrap>", line 1027, in _find_and_load File "<frozen importlib._bootstrap>", line 1006, in _find_and_load_unlocked File "<frozen importlib._bootstrap>", line 688, in _load_unlocked File "<frozen importlib._bootstrap_external>", line 883, in exec_module File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed File "/home/user/app/trellis/models/sparse_structure_flow.py", line 7, in <module> from ..modules.transformer import AbsolutePositionEmbedder, ModulatedTransformerCrossBlock File "/home/user/app/trellis/modules/transformer/__init__.py", line 1, in <module> from .blocks import * File "/home/user/app/trellis/modules/transformer/blocks.py", line 4, in <module> from ..attention import MultiHeadAttention File "/home/user/app/trellis/modules/attention/__init__.py", line 35, in <module> from .full_attn import * File "/home/user/app/trellis/modules/attention/full_attn.py", line 9, in <module> import flash_attn ModuleNotFoundError: No module named 'flash_attn'

Container logs: