Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -57,7 +57,8 @@ pipe = HeliosPyramidPipeline.from_pretrained(
|
|
| 57 |
is_distilled=True
|
| 58 |
)
|
| 59 |
pipe.to("cuda")
|
| 60 |
-
|
|
|
|
| 61 |
|
| 62 |
# ---------------------------------------------------------------------------
|
| 63 |
# Generation (decorated for ZeroGPU)
|
|
|
|
| 57 |
is_distilled=True
|
| 58 |
)
|
| 59 |
pipe.to("cuda")
|
| 60 |
+
|
| 61 |
+
pipe.transformer.set_attention_backend("_flash_3_hub")
|
| 62 |
|
| 63 |
# ---------------------------------------------------------------------------
|
| 64 |
# Generation (decorated for ZeroGPU)
|