Create monkey.py

monkey.py

@@ -0,0 +1,17 @@
+from dataclasses import dataclass, field
+from typing import List, Optional
+
+# Create a fixed version of the TeaCacheConfig class
+@dataclass
+class TeaCacheConfig:
+    l1_threshold: Optional[float] = None
+    skip_layer_identifiers: List[str] = field(default_factory=lambda: ["blocks", "transformer_blocks", 
+                                                                      "single_transformer_blocks", 
+                                                                      "temporal_transformer_blocks"])
+    _polynomial_coefficients: Optional[List[float]] = None
+
+# Apply the monkey patch
+def apply_patches():
+    import diffusers.pipelines.teacache_utils
+    diffusers.pipelines.teacache_utils.TeaCacheConfig = TeaCacheConfig
+    print("Applied monkey patch for TeaCacheConfig")