Update model.py

model.py

@@ -300,7 +300,7 @@ class StreamMultiDiffusion(nn.Module):
                 (b, 4, self.latent_height, self.latent_width), dtype=self.dtype, device=self.device))
         else:
             self.x_t_latent_buffer = torch.zeros(
-                (b, 4, self.latent_height, self.latent_width), dtype=self.dtype, device=self.device))
+                (b, 4, self.latent_height, self.latent_width), dtype=self.dtype, device=self.device)
 
     def reset_state(self) -> None:
         # TODO Reset states for context switch between multiple users.