Spaces:

adamelliotfields
/

diffusion-xl

Running on Zero

adamelliotfields commited on about 1 month ago

Commit

c60d9a3

•

1 Parent(s): b7bdcdb

Fix VAE

Files changed (1) hide show

lib/loader.py CHANGED Viewed

@@ -129,7 +129,7 @@ class Loader:
         if self.vae is None:
             self.vae = AutoencoderKL.from_pretrained(
                 Config.VAE_MODEL,
-                torch_dtype=torch.float32,  # vae is full-precision
             ).to(self.device)
         kwargs["vae"] = self.vae

         if self.vae is None:
             self.vae = AutoencoderKL.from_pretrained(
                 Config.VAE_MODEL,
+                torch_dtype=torch.float16,
             ).to(self.device)
         kwargs["vae"] = self.vae