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)
|