runtime error

Exit code: 1. Reason: ckages/transformers/utils/import_utils.py", line 1390, in _get_module # docstyle-ignore 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 1004, in _find_and_load_unlocked ModuleNotFoundError: No module named 'transformers.models.mega.configuration_mega' The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/user/app/app.py", line 19, in <module> model = AutoModelForCausalLM.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 538, in from_pretrained kwargs["torch_dtype"] = "auto" File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 749, in keys self._model_mapping._model_mapping = self File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 750, in <listcomp> self._extra_content = {} File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 746, in _load_attr_from_module self._config_mapping = config_mapping File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 690, in getattribute_from_module return None File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1380, in __getattr__ File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1392, in _get_module {0} requires the PyTorch library but it was not found in your environment. Checkout the instructions on the RuntimeError: Failed to import transformers.models.mega.configuration_mega because of the following error (look up to see its traceback): No module named 'transformers.models.mega.configuration_mega'

Container logs: