VectorSpaceLab / OmniGen

OmniGen: Unified Image Generation. https://arxiv.org/pdf/2409.11340
MIT License
1.39k stars 67 forks source link

NameError: name 'is_torch_npu_available' is not defined. Did you mean: 'is_torch_xla_available'? #66

Open larini opened 2 hours ago

larini commented 2 hours ago

While running python app.py, I get this error:

Fetching 10 files: 100%|███████████████████████████████████████████████████████████████████████| 10/10 [00:00<?, ?it/s] Loading safetensors Traceback (most recent call last): File "C:\git\OmniGen\app.py", line 9, in pipe = OmniGenPipeline.from_pretrained( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\git\OmniGen\OmniGen\pipeline.py", line 93, in from_pretrained return cls(vae, model, processor) ^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\git\OmniGen\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'?

staoxiao commented 2 hours ago

see https://github.com/VectorSpaceLab/OmniGen/issues/67