stream dropout update

ldf_models/diffusion_forcing_wan.py

@@ -226,12 +226,12 @@ class DiffForcingWanModel(nn.Module):
                     text_list, text_end_list
                 ):
                     if np.random.rand() > self.drop_out:
-                        single_text_list = [""]
-                        single_text_end_list = [0, seq_len]
-                    else:
                         single_text_end_list = [0] + [
                             min(t, seq_len) for t in single_text_end_list
                         ]
+                    else:
+                        single_text_list = [""]
+                        single_text_end_list = [0, seq_len]
                     single_text_length_list = [
                         t - b
                         for t, b in zip(