add firstblockcache hook

app.py

@@ -13,7 +13,7 @@ import tempfile
 import numpy as np
 from PIL import Image
 import random
-
+from diffusers.hooks import apply_first_block_cache, FirstBlockCacheConfig
 from optimization import optimize_pipeline_
 
 
@@ -46,6 +46,8 @@ pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID,
 ).to('cuda')
 
 
+apply_first_block_cache(pipe.transformer, FirstBlockCacheConfig(threshold=0.05))
+
 optimize_pipeline_(pipe,
     image=Image.new('RGB', (LANDSCAPE_WIDTH, LANDSCAPE_HEIGHT)),
     prompt='prompt',