Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -263,11 +263,11 @@ pipe = WanImageToVideoPipeline.from_pretrained(
|
|
| 263 |
).to('cuda')
|
| 264 |
original_scheduler = copy.deepcopy(pipe.scheduler)
|
| 265 |
|
| 266 |
-
if os.path.exists(CACHE_DIR):
|
| 267 |
-
|
| 268 |
-
|
| 269 |
-
else:
|
| 270 |
-
|
| 271 |
|
| 272 |
quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
|
| 273 |
quantize_(pipe.transformer, Float8DynamicActivationFloat8WeightConfig())
|
|
@@ -562,7 +562,7 @@ CSS = """
|
|
| 562 |
with gr.Blocks(theme=gr.themes.Soft(), css=CSS, delete_cache=(3600, 10800)) as demo:
|
| 563 |
gr.Markdown("## WAMU V2 - Wan 2.2 I2V (14B) 🐢🐢")
|
| 564 |
gr.Markdown("#### ℹ️ **A Note on Performance:** This version prioritizes a straightforward setup over maximum speed, so performance may vary.")
|
| 565 |
-
gr.Markdown('Try the
|
| 566 |
gr.Markdown("Run Wan 2.2 in just 4-8 steps, fp8 quantization & AoT compilation - compatible with 🧨 diffusers and ZeroGPU.")
|
| 567 |
|
| 568 |
with gr.Row():
|
|
|
|
| 263 |
).to('cuda')
|
| 264 |
original_scheduler = copy.deepcopy(pipe.scheduler)
|
| 265 |
|
| 266 |
+
# if os.path.exists(CACHE_DIR):
|
| 267 |
+
# shutil.rmtree(CACHE_DIR)
|
| 268 |
+
# print("Deleted Hugging Face cache.")
|
| 269 |
+
# else:
|
| 270 |
+
# print("No hub cache found.")
|
| 271 |
|
| 272 |
quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
|
| 273 |
quantize_(pipe.transformer, Float8DynamicActivationFloat8WeightConfig())
|
|
|
|
| 562 |
with gr.Blocks(theme=gr.themes.Soft(), css=CSS, delete_cache=(3600, 10800)) as demo:
|
| 563 |
gr.Markdown("## WAMU V2 - Wan 2.2 I2V (14B) 🐢🐢")
|
| 564 |
gr.Markdown("#### ℹ️ **A Note on Performance:** This version prioritizes a straightforward setup over maximum speed, so performance may vary.")
|
| 565 |
+
gr.Markdown('Try the alternative version: [WAMU space](https://huggingface.co/spaces/r3gm/wan2-2-fp8da-aoti-preview2)')
|
| 566 |
gr.Markdown("Run Wan 2.2 in just 4-8 steps, fp8 quantization & AoT compilation - compatible with 🧨 diffusers and ZeroGPU.")
|
| 567 |
|
| 568 |
with gr.Row():
|