runtime error

Exit code: 1. Reason: 5:39<05:39, 339.62s/it] Loading checkpoint shards: 100%|██████████| 2/2 [07:50<00:00, 217.00s/it] Loading checkpoint shards: 100%|██████████| 2/2 [07:50<00:00, 235.41s/it] generation_config.json: 0%| | 0.00/116 [00:00<?, ?B/s] generation_config.json: 100%|██████████| 116/116 [00:00<00:00, 267kB/s] adapter_model.safetensors: 0%| | 0.00/33.6M [00:00<?, ?B/s] adapter_model.safetensors: 100%|██████████| 33.6M/33.6M [00:01<00:00, 29.0MB/s] adapter_model.safetensors: 100%|██████████| 33.6M/33.6M [00:01<00:00, 29.0MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 39, in <module> model = AutoModelForCausalLM.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 600, in from_pretrained return model_class.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 317, in _wrapper return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 5177, in from_pretrained model.load_adapter( File "/usr/local/lib/python3.10/site-packages/transformers/integrations/peft.py", line 307, in load_adapter self._dispatch_accelerate_model( File "/usr/local/lib/python3.10/site-packages/transformers/integrations/peft.py", line 558, in _dispatch_accelerate_model dispatch_model( File "/usr/local/lib/python3.10/site-packages/accelerate/big_modeling.py", line 383, in dispatch_model raise ValueError( ValueError: We need an `offload_dir` to dispatch this model according to this `device_map`, the following submodules need to be offloaded: model.layers.19, model.layers.20, model.layers.21, model.layers.22, model.layers.23, model.layers.24, model.layers.25, model.layers.26, model.layers.27, model.layers.28, model.layers.29, model.layers.30, model.layers.31, model.norm, model.rotary_emb, lm_head.

Container logs: