flux series vram management

2026-03-23 17:38:10 +00:00 · 2025-07-15 20:11:02 +08:00
parent cbd10fb27d
commit af6b1d4246
16 changed files with 629 additions and 27 deletions
--- a/diffsynth/models/flux_infiniteyou.py
+++ b/diffsynth/models/flux_infiniteyou.py
@@ -104,6 +104,7 @@ class InfiniteYouImageProjector(nn.Module):
    def forward(self, x):

        latents = self.latents.repeat(x.size(0), 1, 1)
+        latents = latents.to(dtype=x.dtype, device=x.device)

        x = self.proj_in(x)