from_tf=True

app.py

@@ -373,7 +373,8 @@ class PretrainVae:
         from diffusers import AutoencoderKL, DiffusionPipeline
         self.vae = AutoencoderKL.from_pretrained("gsdf/Counterfeit-V2.5",  # segmind/small-sd
                                                  subfolder="vae",
-                                                 cache_dir="./vae/pretrain_vae").to(device)
+                                                 cache_dir="./vae/pretrain_vae",
+                                                 from_tf=True).to(device)
         self.vae.requires_grad_(False)
         self.middle_c = 4
         self.vae_scaleing = 0.18215