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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +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", torch_dtype=dtype, device_map='cuda').to(device)
31
 
32
  # Load the relight LoRA
33
  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).to(device)
31
 
32
  # Load the relight LoRA
33
  pipe.load_lora_weights(