Update model.py

model.py

@@ -72,6 +72,7 @@ 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