inference

This commit is contained in:
Haoming 2026-02-21 03:29:51 +08:00
parent a481654fe7
commit 24279f0894

View File

@ -795,6 +795,7 @@ class Anima(MiniTrainDIT):
super().__init__(*args, **kwargs)
self.llm_adapter = LLMAdapter()
@torch.inference_mode()
def preprocess_text_embeds(self, text_embeds, text_ids):
if text_ids is not None:
device = self.llm_adapter.embed.weight.device