Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
19 |
pipe = QwenImageEditPipeline.from_pretrained("Qwen/Qwen-Image-Edit", torch_dtype=dtype).to(device)
|
20 |
|
21 |
# --- Ahead-of-time compilation ---
|
22 |
-
|
23 |
|
24 |
# --- UI Constants and Helpers ---
|
25 |
MAX_SEED = np.iinfo(np.int32).max
|
|
|
19 |
pipe = QwenImageEditPipeline.from_pretrained("Qwen/Qwen-Image-Edit", torch_dtype=dtype).to(device)
|
20 |
|
21 |
# --- Ahead-of-time compilation ---
|
22 |
+
optimize_pipeline_(pipe, image=Image.new("RGB", (1024, 1024)), prompt='prompt')
|
23 |
|
24 |
# --- UI Constants and Helpers ---
|
25 |
MAX_SEED = np.iinfo(np.int32).max
|