Qwen-Image-Edit-next-scene

Runtime error

mrbui1990 commited on 26 days ago

Commit

4d07b58

verified ·

1 Parent(s): a487b94

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -382,7 +382,7 @@ dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509",
-                                                transformer= QwenImageTransformer2DModel.from_pretrained("linoyts/Qwen-Image-Edit-Rapid-AIO",
                                                                                                          subfolder='transformer',
                                                                                                          torch_dtype=dtype,
                                                                                                          device_map='cuda'),torch_dtype=dtype).to(device)

 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509",
+                                                transformer= QwenImageTransformer2DModel.from_pretrained("linoyts/Qwen-Image-Edit-Rapid-AIO-v10",
                                                                                                          subfolder='transformer',
                                                                                                          torch_dtype=dtype,
                                                                                                          device_map='cuda'),torch_dtype=dtype).to(device)