mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-07-21 21:01:24 +08:00
inference
This commit is contained in:
parent
a481654fe7
commit
24279f0894
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user