(self, num_inference_steps, rel_l1_thresh)
| 611 | |
| 612 | class TeaCache: |
| 613 | def __init__(self, num_inference_steps, rel_l1_thresh): |
| 614 | self.num_inference_steps = num_inference_steps |
| 615 | self.step = 0 |
| 616 | self.accumulated_rel_l1_distance = 0 |
| 617 | self.previous_modulated_input = None |
| 618 | self.rel_l1_thresh = rel_l1_thresh |
| 619 | self.previous_residual = None |
| 620 | self.previous_hidden_states = None |
| 621 | |
| 622 | def check(self, dit: FluxDiT, hidden_states, conditioning): |
| 623 | inp = hidden_states.clone() |