r3gm commited on
Commit
93dfba4
·
verified ·
1 Parent(s): 153a11a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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
- print("Loading...")
43
- subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
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 10 + gen_time
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():