Update model.py

model.py

@@ -81,7 +81,7 @@ class StreamMultiDiffusion(nn.Module):
             bootstrap_steps > torch.arange(len(t_index_list))).to(dtype=self.dtype, device=self.device)
         self.bootstrap_mix_steps = bootstrap_mix_steps
         self.bootstrap_mix_ratios = (
-            bootstrap_mix_steps - torch.arange(len(t_index_list), dtype=self.dtype, device=self.device)).clip_(0, 1)
+            bootstrap_mix_steps - torch.arange(len(t_index_list), device=self.device)).clip_(0, 1).to(self.dtype)
         # self.bootstrap_leak_sensitivity = bootstrap_leak_sensitivity
         self.preprocess_mask_cover_alpha = preprocess_mask_cover_alpha
         self.mask_type = mask_type