diff --git a/TeaCache_Lumina2.py b/TeaCache_Lumina2.py index 685969a..597d6ef 100644 --- a/TeaCache_Lumina2.py +++ b/TeaCache_Lumina2.py @@ -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,) \ No newline at end of file + return (new_model,)