multimodalart HF Staff commited on
Commit
741b852
·
verified ·
1 Parent(s): efd7824

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -27,11 +27,7 @@ from huggingface_hub import InferenceClient
27
  dtype = torch.bfloat16
28
  device = "cuda" if torch.cuda.is_available() else "cpu"
29
 
30
- pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509",
31
- #transformer= QwenImageTransformer2DModel.from_pretrained("linoyts/Qwen-Image-Edit-Rapid-AIO",
32
- # subfolder='transformer',
33
- torch_dtype=dtype,
34
- device_map='cuda'),torch_dtype=dtype).to(device)
35
 
36
  # Load the relight LoRA
37
  pipe.load_lora_weights(
 
27
  dtype = torch.bfloat16
28
  device = "cuda" if torch.cuda.is_available() else "cpu"
29
 
30
+ pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509", torch_dtype=dtype, device_map='cuda').to(device)
 
 
 
 
31
 
32
  # Load the relight LoRA
33
  pipe.load_lora_weights(