| 554 | dim=1) |
| 555 | |
| 556 | def teacache_init( |
| 557 | self, |
| 558 | use_ret_steps=True, |
| 559 | teacache_thresh=0.2, |
| 560 | sample_steps=40, |
| 561 | model_scale='multitalk-480', |
| 562 | ): |
| 563 | print("teacache_init") |
| 564 | self.enable_teacache = True |
| 565 | |
| 566 | self.__class__.cnt = 0 |
| 567 | self.__class__.num_steps = sample_steps*3 |
| 568 | self.__class__.teacache_thresh = teacache_thresh |
| 569 | self.__class__.accumulated_rel_l1_distance_even = 0 |
| 570 | self.__class__.accumulated_rel_l1_distance_odd = 0 |
| 571 | self.__class__.previous_e0_even = None |
| 572 | self.__class__.previous_e0_odd = None |
| 573 | self.__class__.previous_residual_even = None |
| 574 | self.__class__.previous_residual_odd = None |
| 575 | self.__class__.use_ret_steps = use_ret_steps |
| 576 | |
| 577 | if use_ret_steps: |
| 578 | if model_scale == 'multitalk-480': |
| 579 | self.__class__.coefficients = [ 2.57151496e+05, -3.54229917e+04, 1.40286849e+03, -1.35890334e+01, 1.32517977e-01] |
| 580 | if model_scale == 'multitalk-720': |
| 581 | self.__class__.coefficients = [ 8.10705460e+03, 2.13393892e+03, -3.72934672e+02, 1.66203073e+01, -4.17769401e-02] |
| 582 | self.__class__.ret_steps = 5*3 |
| 583 | self.__class__.cutoff_steps = sample_steps*3 |
| 584 | else: |
| 585 | if model_scale == 'multitalk-480': |
| 586 | self.__class__.coefficients = [-3.02331670e+02, 2.23948934e+02, -5.25463970e+01, 5.87348440e+00, -2.01973289e-01] |
| 587 | |
| 588 | if model_scale == 'multitalk-720': |
| 589 | self.__class__.coefficients = [-114.36346466, 65.26524496, -18.82220707, 4.91518089, -0.23412683] |
| 590 | self.__class__.ret_steps = 1*3 |
| 591 | self.__class__.cutoff_steps = sample_steps*3 - 3 |
| 592 | print("teacache_init done") |
| 593 | |
| 594 | def disable_teacache(self): |
| 595 | self.enable_teacache = False |