Update app.py
Browse files
app.py
CHANGED
|
@@ -38,9 +38,9 @@ os.environ["TOKENIZERS_PARALLELISM"] = "true"
|
|
| 38 |
warnings.filterwarnings("ignore")
|
| 39 |
IS_ZERO_GPU = bool(os.getenv("SPACES_ZERO_GPU"))
|
| 40 |
|
| 41 |
-
if IS_ZERO_GPU:
|
| 42 |
-
|
| 43 |
-
|
| 44 |
|
| 45 |
# --- FRAME EXTRACTION JS & LOGIC ---
|
| 46 |
|
|
@@ -373,7 +373,7 @@ def get_inference_duration(
|
|
| 373 |
inter_time = (total_out_frames * 0.02)
|
| 374 |
gen_time += inter_time
|
| 375 |
|
| 376 |
-
return
|
| 377 |
|
| 378 |
|
| 379 |
@spaces.GPU(duration=get_inference_duration)
|
|
@@ -563,7 +563,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=CSS, delete_cache=(3600, 10800)) as d
|
|
| 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 previous version: [WAMU v1](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():
|
| 569 |
with gr.Column():
|
|
|
|
| 38 |
warnings.filterwarnings("ignore")
|
| 39 |
IS_ZERO_GPU = bool(os.getenv("SPACES_ZERO_GPU"))
|
| 40 |
|
| 41 |
+
# if IS_ZERO_GPU:
|
| 42 |
+
# print("Loading...")
|
| 43 |
+
# subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
|
| 44 |
|
| 45 |
# --- FRAME EXTRACTION JS & LOGIC ---
|
| 46 |
|
|
|
|
| 373 |
inter_time = (total_out_frames * 0.02)
|
| 374 |
gen_time += inter_time
|
| 375 |
|
| 376 |
+
return 15 + gen_time
|
| 377 |
|
| 378 |
|
| 379 |
@spaces.GPU(duration=get_inference_duration)
|
|
|
|
| 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 previous version: [WAMU v1](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():
|
| 569 |
with gr.Column():
|