update

OmniGen/pipeline.py

@@ -60,6 +60,8 @@ class OmniGenPipeline:
 
     @classmethod
     def from_pretrained(cls, model_name, vae_path: str=None):
+        device = devicetorch.get(torch)
+        print(f">Device={device}")
         if not os.path.exists(model_name):
             logger.info("Model not found, downloading...")
             cache_folder = os.getenv('HF_HUB_CACHE')