From 24279f0894be79354efcf653bca69c86d31ec693 Mon Sep 17 00:00:00 2001 From: Haoming Date: Sat, 21 Feb 2026 03:29:51 +0800 Subject: [PATCH] inference --- backend/nn/anima.py | 1 + 1 file changed, 1 insertion(+) diff --git a/backend/nn/anima.py b/backend/nn/anima.py index ec754171..9d0ba1c7 100644 --- a/backend/nn/anima.py +++ b/backend/nn/anima.py @@ -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