Hugging Face's logo

Spaces:
Shitao
/
OmniGen
Running on Zero

App Files Community
11
Shitao commited on
Commit
8032ab8
•
1 Parent(s): 53aed50

Update OmniGen/pipeline.py

Browse files
Files changed (1) hide show
  1. OmniGen/pipeline.py +0 -2
OmniGen/pipeline.py CHANGED
@@ -58,8 +58,6 @@ class OmniGenPipeline:
58
  self.device = torch.device("cuda")
59
  elif torch.backends.mps.is_available():
60
  self.device = torch.device("mps")
61
- elif is_torch_npu_available():
62
- self.device = torch.device("npu")
63
  else:
64
  logger.info("Don't detect any available devices, using CPU instead")
65
  self.device = torch.device("cpu")
 
58
  self.device = torch.device("cuda")
59
  elif torch.backends.mps.is_available():
60
  self.device = torch.device("mps")
 
 
61
  else:
62
  logger.info("Don't detect any available devices, using CPU instead")
63
  self.device = torch.device("cpu")