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