Update handler.py

handler.py

@@ -129,13 +129,18 @@ class EndpointHandler:
             # https://github.com/huggingface/diffusers/pull/10163/files#diff-777f4ee62cb325371233a450e0f6cc0ba357a3fade2ec2dea912260b4f8d08ceR67-R74
             
             faster_cache_config = FasterCacheConfig(
+                current_timestep_callback=lambda: self.text_to_video.current_timestep,
                 spatial_attention_block_skip_range=2,
                 spatial_attention_timestep_skip_range=(-1, 901),
                 unconditional_batch_skip_range=2,
                 attention_weight_callback=lambda _: 0.5,
                 is_guidance_distilled=True,
+
+                # do we need to uncomment those?
+                #unconditional_batch_timestep_skip_range=(-1, 901),
+                #tensor_format="BFCHW",
             )
-        
+            
             self.text_to_video.transformer.enable_cache(faster_cache_config)