runtime error

Exit code: 1. Reason: lorence2.py . Make sure to double-check they do not contain any added malicious code. To avoid downloading new versions of the code file, you can pin a revision. modeling_florence2.py: 0%| | 0.00/127k [00:00<?, ?B/s] modeling_florence2.py: 100%|██████████| 127k/127k [00:00<00:00, 6.03MB/s] Traceback (most recent call last): File "/home/user/app/src/florence/model.py", line 27, in <module> AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, trust_remote_code=True) File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 551, in from_pretrained model_class = get_class_from_dynamic_module( File "/usr/local/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 502, in get_class_from_dynamic_module final_module = get_cached_module_file( File "/usr/local/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 327, in get_cached_module_file modules_needed = check_imports(resolved_module_file) File "/usr/local/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 182, in check_imports raise ImportError( ImportError: This modeling file requires the following packages that were not found in your environment: flash_attn. Run `pip install flash_attn` The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/user/app/app.py", line 7, in <module> from src.florence.task import ocr_task File "/home/user/app/src/florence/task.py", line 10, in <module> from src.florence.model import run_example File "/home/user/app/src/florence/model.py", line 37, in <module> raise CustomExceptionHandling(e, sys) from e src.exception.CustomExceptionHandling: Error occured in python script name [/home/user/app/src/florence/model.py] line number [27] error message[This modeling file requires the following packages that were not found in your environment: flash_attn. Run `pip install flash_attn`]

Container logs: