accelerate和diffusers

app.py

@@ -373,8 +373,7 @@ 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",
-                                                 from_tf=True).to(device)
+                                                 cache_dir="./vae/pretrain_vae").to(device)
         self.vae.requires_grad_(False)
         self.middle_c = 4
         self.vae_scaleing = 0.18215

requirements.txt

@@ -4,4 +4,5 @@ huggingface-hub==0.16.4
 numpy==1.23.4
 torch
 tqdm
-diffusers
+diffusers==0.20.1
+accelerate