runtime error

Exit code: 1. Reason: | 68.9M/5.48G [00:03<03:05, 29.2MB/s] Downloading (incomplete total...): 11%|█ | 605M/5.48G [00:04<00:22, 213MB/s]  Downloading (incomplete total...): 19%|█▊ | 1.02G/5.48G [00:06<00:20, 221MB/s] Fetching 15 files: 27%|██▋ | 4/15 [00:06<00:17, 1.61s/it] Downloading (incomplete total...): 30%|██▉ | 1.63G/5.48G [00:07<00:11, 326MB/s] Downloading (incomplete total...): 69%|██████▉ | 3.79G/5.48G [00:08<00:02, 834MB/s] Downloading (incomplete total...): 100%|██████████| 5.48G/5.48G [00:09<00:00, 1.02GB/s] Fetching 15 files: 87%|████████▋ | 13/15 [00:09<00:01, 1.60it/s] Fetching 15 files: 100%|██████████| 15/15 [00:09<00:00, 1.61it/s] Download complete: 100%|██████████| 5.48G/5.48G [00:09<00:00, 1.02GB/s]  Loading pipeline components...: 0%| | 0/7 [00:00<?, ?it/s] Download complete: 100%|██████████| 5.48G/5.48G [00:09<00:00, 583MB/s] Loading weights: 0%| | 0/196 [00:00<?, ?it/s] Loading weights: 100%|██████████| 196/196 [00:00<00:00, 1187.15it/s] Loading weights: 0%| | 0/396 [00:00<?, ?it/s] Loading weights: 100%|██████████| 396/396 [00:00<00:00, 1450.05it/s] Loading pipeline components...: 100%|██████████| 7/7 [00:01<00:00, 6.39it/s] Loading pipeline components...: 100%|██████████| 7/7 [00:01<00:00, 6.39it/s] Inference Mode: All SD components frozen Traceback (most recent call last): File "/app/app.py", line 23, in <module> model, config = load_model() ~~~~~~~~~~^^ File "/app/app.py", line 16, in load_model model = Audio2ImageModel(config, load_sd=True) File "/app/main2.py", line 406, in __init__ clap_text_dim = t.shape[-1] ^^^^^^^ AttributeError: 'BaseModelOutputWithPooling' object has no attribute 'shape'

Container logs: