Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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(
|