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