Qwen-Image-Layered

Runtime error

naykun commited on 5 days ago

Commit

20eb1ad

1 Parent(s): 59bff29

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,9 +17,10 @@ MAX_SEED = np.iinfo(np.int32).max
 from huggingface_hub import login
 login(token=os.environ.get('hf'))
-pipeline = QwenImageLayeredPipeline.from_pretrained("Qwen/Qwen-Image-Layered")
-pipeline = pipeline.to("cuda", torch.bfloat16)
-pipeline.set_progress_bar_config(disable=None)
 def ensure_dirname(path: str):
     if path and not os.path.exists(path):

 from huggingface_hub import login
 login(token=os.environ.get('hf'))
+dtype = torch.bfloat16
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipeline = QwenImageLayeredPipeline.from_pretrained("Qwen/Qwen-Image-Layered", torch_dtype=dtype).to(device)
+# pipeline.set_progress_bar_config(disable=None)
 def ensure_dirname(path: str):
     if path and not os.path.exists(path):