Spaces:

Alpha-VLLM
/

Lumina-Next-T2I

Running on Zero

PommesPeter commited on May 29

Commit

7a92cca

•

1 Parent(s): 64f5019

Update models/model.py

Files changed (1) hide show

models/model.py CHANGED Viewed

@@ -862,7 +862,7 @@ class NextDiT(nn.Module):
                 x, _ = x.chunk(2, dim=1)
             else:
                 x = [_.chunk(2, dim=0)[0] for _ in x]
-        return x`
     def forward_with_cfg(
         self,

                 x, _ = x.chunk(2, dim=1)
             else:
                 x = [_.chunk(2, dim=0)[0] for _ in x]
+        return x
     def forward_with_cfg(
         self,