add new model

app.py

@@ -62,6 +62,7 @@ class VAETester:
             "stable-diffusion-3-medium": AutoencoderKL.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", subfolder="vae").to(self.device),
             "FLUX.1": AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae").to(self.device),
             "CogView4-6B": AutoencoderKL.from_pretrained("THUDM/CogView4-6B", subfolder="vae").to(self.device),
+            "playground-v2.5": AutoencoderKL.from_pretrained("playgroundai/playground-v2.5-1024px-aesthetic", subfolder="vae").to(self.device),
         }
         # Define the desired order of models
         order = [
@@ -72,6 +73,7 @@ class VAETester:
             #"sd-vae-ft-mse (remote)",
             "sdxl-vae",
             #"sdxl-vae (remote)",
+            "playground-v2.5",
             "stable-diffusion-3-medium",
             "FLUX.1",
             #"FLUX.1 (remote)",
@@ -196,7 +198,7 @@ with gr.Blocks(title="VAE Performance Tester", css=".monospace-text {font-family
             with gr.Row():
                 diff_gallery = gr.Gallery(label="Difference Maps", columns=4, height=512)
                 recon_gallery = gr.Gallery(label="Reconstructed Images", columns=4, height=512)
-            scores_output = gr.Textbox(label="Sum of differences (lower is better) | Processing time (lower is faster)", lines=9, elem_classes="monospace-text")
+            scores_output = gr.Textbox(label="Sum of differences (lower is better) | Processing time (lower is faster)", lines=10, elem_classes="monospace-text")
 
     if examples:
         with gr.Row():