runtime error

Exit code: 1. Reason: odel.safetensors: 77%|███████▋ | 12.0G/15.5G [00:25<00:06, 512MB/s] model.safetensors: 81%|████████ | 12.6G/15.5G [00:26<00:05, 539MB/s] model.safetensors: 85%|████████▍ | 13.1G/15.5G [00:27<00:04, 529MB/s] model.safetensors: 88%|████████▊ | 13.7G/15.5G [00:28<00:03, 539MB/s] model.safetensors: 92%|█████████▏| 14.2G/15.5G [00:29<00:02, 523MB/s] model.safetensors: 96%|█████████▌| 14.9G/15.5G [00:30<00:01, 559MB/s] model.safetensors: 100%|█████████▉| 15.5G/15.5G [00:31<00:00, 495MB/s] special_tokens_map.json: 0%| | 0.00/670 [00:00<?, ?B/s] special_tokens_map.json: 100%|██████████| 670/670 [00:00<00:00, 4.96MB/s] tokenizer.json: 0%| | 0.00/1.85M [00:00<?, ?B/s] tokenizer.json: 100%|██████████| 1.85M/1.85M [00:00<00:00, 13.5MB/s] tokenizer_config.json: 0%| | 0.00/10.1k [00:00<?, ?B/s] tokenizer_config.json: 100%|██████████| 10.1k/10.1k [00:00<00:00, 61.3MB/s] vae/config.json: 0%| | 0.00/607 [00:00<?, ?B/s] vae/config.json: 100%|██████████| 607/607 [00:00<00:00, 6.09MB/s] diffusion_pytorch_model.safetensors: 0%| | 0.00/335M [00:00<?, ?B/s] diffusion_pytorch_model.safetensors: 100%|█████████▉| 335M/335M [00:01<00:00, 287MB/s] diffusion_pytorch_model.safetensors: 100%|█████████▉| 335M/335M [00:01<00:00, 278MB/s] Loading safetensors Traceback (most recent call last): File "/home/user/app/app.py", line 9, in <module> pipe = OmniGenPipeline.from_pretrained( File "/home/user/app/OmniGen/pipeline.py", line 93, in from_pretrained return cls(vae, model, processor) File "/home/user/app/OmniGen/pipeline.py", line 61, in __init__ elif is_torch_npu_available(): NameError: name 'is_torch_npu_available' is not defined. Did you mean: 'is_torch_xla_available'?

Container logs: