r3gm commited on
Commit
6373d9e
·
verified ·
1 Parent(s): c5d7a47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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
- 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,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 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():
 
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():