Spaces:
Running
on
Zero
Running
on
Zero
add new models
Browse files
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=
|
|
|
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)
|