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