runtime error

Exit code: 1. Reason: 00:00<00:00, 126MB/s] flash_attn_triton.py: 0%| | 0.00/28.2k [00:00<?, ?B/s] flash_attn_triton.py: 100%|██████████| 28.2k/28.2k [00:00<00:00, 82.1MB/s] Encountered exception while importing triton_pre_mlir: No module named 'triton_pre_mlir' Traceback (most recent call last): File "/home/user/app/app.py", line 31, in <module> m = AutoModelForCausalLM.from_pretrained( File "/home/user/.pyenv/versions/3.10.17/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 531, in from_pretrained config, kwargs = AutoConfig.from_pretrained( File "/home/user/.pyenv/versions/3.10.17/lib/python3.10/site-packages/transformers/models/auto/configuration_auto.py", line 1123, in from_pretrained config_class = get_class_from_dynamic_module( File "/home/user/.pyenv/versions/3.10.17/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 558, in get_class_from_dynamic_module final_module = get_cached_module_file( File "/home/user/.pyenv/versions/3.10.17/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 421, in get_cached_module_file get_cached_module_file( File "/home/user/.pyenv/versions/3.10.17/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 421, in get_cached_module_file get_cached_module_file( File "/home/user/.pyenv/versions/3.10.17/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 421, in get_cached_module_file get_cached_module_file( File "/home/user/.pyenv/versions/3.10.17/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 383, in get_cached_module_file modules_needed = check_imports(resolved_module_file) File "/home/user/.pyenv/versions/3.10.17/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 215, in check_imports raise ImportError( ImportError: This modeling file requires the following packages that were not found in your environment: triton_pre_mlir. Run `pip install triton_pre_mlir`

Container logs: