rizavelioglu commited on
Commit
7069342
·
verified ·
1 Parent(s): a766367

add new models

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -55,15 +55,19 @@ class VAETester:
55
  """Load configurations for local and remote VAE models"""
56
  local_vaes = {
57
  "stable-diffusion-v1-4": AutoencoderKL.from_pretrained("CompVis/stable-diffusion-v1-4", subfolder="vae").to(self.device),
 
 
58
  "sd-vae-ft-mse": AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(self.device),
59
  "sdxl-vae": AutoencoderKL.from_pretrained("stabilityai/sdxl-vae").to(self.device),
60
  "stable-diffusion-3-medium": AutoencoderKL.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", subfolder="vae").to(self.device),
61
  "FLUX.1": AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae").to(self.device),
62
- "CogView4-6B": AutoencoderKL.from_pretrained("THUDM/CogView4-6B", subfolder="vae").to(self.device),
63
  }
64
  # Define the desired order of models
65
  order = [
66
  "stable-diffusion-v1-4",
 
 
67
  "sd-vae-ft-mse",
68
  #"sd-vae-ft-mse (remote)",
69
  "sdxl-vae",
@@ -205,4 +209,4 @@ with gr.Blocks(title="VAE Performance Tester", css=".monospace-text {font-family
205
  )
206
 
207
  if __name__ == "__main__":
208
- demo.launch(share=True, ssr_mode=False)
 
55
  """Load configurations for local and remote VAE models"""
56
  local_vaes = {
57
  "stable-diffusion-v1-4": AutoencoderKL.from_pretrained("CompVis/stable-diffusion-v1-4", subfolder="vae").to(self.device),
58
+ "eq-vae-ema": AutoencoderKL.from_pretrained("zelaki/eq-vae-ema").to(self.device),
59
+ "eq-sdxl-vae": AutoencoderKL.from_pretrained("KBlueLeaf/EQ-SDXL-VAE").to(self.device),
60
  "sd-vae-ft-mse": AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(self.device),
61
  "sdxl-vae": AutoencoderKL.from_pretrained("stabilityai/sdxl-vae").to(self.device),
62
  "stable-diffusion-3-medium": AutoencoderKL.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", subfolder="vae").to(self.device),
63
  "FLUX.1": AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae").to(self.device),
64
+ "CogView4-6B": AutoencoderKL.from_pretrained("THUDM/CogView4-6B", subfolder="vae").to(self.device),
65
  }
66
  # Define the desired order of models
67
  order = [
68
  "stable-diffusion-v1-4",
69
+ "eq-vae-ema",
70
+ "eq-sdxl-vae",
71
  "sd-vae-ft-mse",
72
  #"sd-vae-ft-mse (remote)",
73
  "sdxl-vae",
 
209
  )
210
 
211
  if __name__ == "__main__":
212
+ demo.launch(share=False, ssr_mode=False)