Update app.py

app.py

@@ -566,9 +566,13 @@ def infer(
                         
                         print('use_Tiny_VAE = ', use_Tiny_VAE)
 
+                        f Tiny_VAE == "sd-vae-ft-mse":
+                            VAE_id = "stabilityai/sd-vae-ft-mse"
+                        
                         generator = torch.Generator(device).manual_seed(seed)
 
-                        vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch_dtype)
+                        #vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch_dtype)
+                        vae = AutoencoderKL.from_pretrained(VAE_id, torch_dtype=torch_dtype)
 
                         pipe_Tiny_VAE = StableDiffusionPipeline.from_pretrained(model_default, vae=vae, torch_dtype=torch_dtype).to(device)
 
@@ -880,16 +884,6 @@ with gr.Blocks(css=css) as demo:
 
         # Tiny_VAE -----------------------------------------------------------------------------------------------------
         # Checkbox для Tiny_VAE
-        
-        # with gr.Blocks():
-        #     use_Tiny_VAE = gr.Checkbox(
-        #         label="Use Tiny_VAE",
-        #         value=False,
-        #         interactive=True
-        #     )
-
-
-
         with gr.Blocks():
             with gr.Row():
                 use_Tiny_VAE = gr.Checkbox(