runtime error

Exit code: 1. Reason: None of PyTorch, TensorFlow >= 2.0, or Flax have been found. Models won't be available and only tokenizers, configuration and file/data utilities can be used. Traceback (most recent call last): File "/home/user/app/app.py", line 72, in <module> model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", load_in_8bit=True) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1651, in __getattribute__ requires_backends(cls, cls._backends) File "/usr/local/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1639, in requires_backends raise ImportError("".join(failed)) ImportError: AutoModelForCausalLM requires the PyTorch library but it was not found in your environment. Checkout the instructions on the installation page: https://pytorch.org/get-started/locally/ and follow the ones that match your environment. Please note that you may need to restart your runtime after installation.

Container logs: