Update TeaCache_Lumina2.py

2025-06-04 22:01:43 +08:00
parent cdbb9af3e5
commit c44cf0b54f
1 changed files with 3 additions and 4 deletions
@@ -3,10 +3,9 @@ import numpy as np
 from comfy.ldm.common_dit import pad_to_patch_size  # noqa
 from unittest.mock import patch

-
 # referenced from https://github.com/spawner1145/TeaCache/blob/main/TeaCache4Lumina2/teacache_lumina2.py
-# transplanted by @fexli
-# retransplanted by @spawner1145
+# firstly transplanted by @fexli https://github.com/fexli
+# retransplanted by @spawner1145 https://github.com/spawner1145
 def teacache_forward_working(
        self, x, timesteps, context, num_tokens, attention_mask=None, transformer_options={}, **kwargs
 ):
@@ -222,4 +221,4 @@ class TeaCache_Lumina2:
                return model_function(input_val, timestep, **c_condition_dict)

        new_model.set_model_unet_function_wrapper(unet_wrapper_function)
-        return (new_model,)
+        return (new_model,)