Update model.py

model.py

@@ -72,7 +72,6 @@ class StreamMultiDiffusion(nn.Module):
         self.dtype = dtype
         self.seed = seed
         self.sd_version = sd_version
-        print(self.dtype, self.device)
 
         self.autoflush = autoflush
         self.default_mask_std = default_mask_std
@@ -154,7 +153,7 @@ class StreamMultiDiffusion(nn.Module):
             self.pipe.enable_xformers_memory_efficient_attention()
 
         self.vae = (
-            AutoencoderTiny.from_pretrained('madebyollin/taesd').to(device=self.device, dtype=self.dtype)
+            AutoencoderTiny.from_pretrained('madebyollin/taesd', torch_dtype=self.dtype).to(device=self.device)
             if use_tiny_vae else self.pipe.vae
         )
         # self.tokenizer = self.pipe.tokenizer