Update model.py

model.py

@@ -314,7 +314,7 @@ class StreamMultiDiffusion(nn.Module):
             sub_timesteps.append(self.timesteps[t])
         sub_timesteps_tensor = torch.tensor(sub_timesteps, dtype=torch.long, device=self.device)
         if not hasattr(self, 'sub_timesteps_tensor'):
-            self.register_buffer('sub_timesteps_tensor', sub_timesteps_tensor.repeat_interleave(self.frame_bff_size, dim=0)
+            self.register_buffer('sub_timesteps_tensor', sub_timesteps_tensor.repeat_interleave(self.frame_bff_size, dim=0))
         else:
             self.sub_timesteps_tensor = sub_timesteps_tensor.repeat_interleave(self.frame_bff_size, dim=0)