convert preview vae

painebenjamin · Jan 19, 2024 · ffd763d · ffd763d
1 parent dcbbc03
commit ffd763d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/python/enfugue/diffusion/manager.py b/src/python/enfugue/diffusion/manager.py
@@ -4385,7 +4385,7 @@ def get_vae_preview(self, use_xl: bool) -> AutoencoderTiny:
             AutoencoderTiny._dict_from_json_file(vae_config)
         )
         vae_model.load_state_dict(load_state_dict(path), strict=False)
-        return vae_model.to(self.device)
+        return vae_model.to(device=self.device, dtype=self.dtype)
 
     def get_scheduler_class(
         self,