runtime error

Exit code: 1. Reason:  model.safetensors: 80%|████████ | 12.4G/15.5G [00:27<00:06, 496MB/s] model.safetensors: 84%|████████▎ | 12.9G/15.5G [00:28<00:05, 483MB/s] model.safetensors: 87%|████████▋ | 13.5G/15.5G [00:29<00:04, 487MB/s] model.safetensors: 90%|█████████ | 14.0G/15.5G [00:30<00:03, 483MB/s] model.safetensors: 93%|█████████▎| 14.5G/15.5G [00:31<00:02, 479MB/s] model.safetensors: 97%|█████████▋| 15.1G/15.5G [00:32<00:00, 513MB/s] model.safetensors: 100%|█████████▉| 15.5G/15.5G [00:33<00:00, 461MB/s] special_tokens_map.json: 0%| | 0.00/670 [00:00<?, ?B/s] special_tokens_map.json: 100%|██████████| 670/670 [00:00<00:00, 5.55MB/s] tokenizer.json: 0%| | 0.00/1.85M [00:00<?, ?B/s] tokenizer.json: 100%|██████████| 1.85M/1.85M [00:00<00:00, 40.2MB/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, 45.3MB/s] config.json: 0%| | 0.00/607 [00:00<?, ?B/s] config.json: 100%|██████████| 607/607 [00:00<00:00, 4.55MB/s] diffusion_pytorch_model.safetensors: 0%| | 0.00/335M [00:00<?, ?B/s] diffusion_pytorch_model.safetensors: 94%|█████████▎| 314M/335M [00:01<00:00, 306MB/s] diffusion_pytorch_model.safetensors: 100%|█████████▉| 335M/335M [00:01<00:00, 306MB/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: