Update app.py

app.py

@@ -96,7 +96,7 @@ def load_and_prepare_model(model_id):
     #vae = AutoencoderKL.from_pretrained('cross-attention/asymmetric-autoencoder-kl-x-2',use_safetensors=False)
     #vae = AutoencoderKL.from_single_file('https://huggingface.co/ford442/sdxl-vae-bf16/mySLR/myslrVAE_v10.safetensors')
     #vaeX = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse",use_safetensors=True)
-    vaeX = AutoencoderKL.from_pretrained('ford442/Juggernaut-XI-v11-fp32',subfolder='vae', safety_checker=None, use_safetensors=False)   #  ,use_safetensors=True  FAILS
+    #vaeX = AutoencoderKL.from_pretrained('ford442/Juggernaut-XI-v11-fp32',subfolder='vae', safety_checker=None, use_safetensors=False)   #  ,use_safetensors=True  FAILS
     #vaeX = AutoencoderKL.from_pretrained('ford442/RealVisXL_V5.0_FP64',subfolder='vae').to(torch.bfloat16)   #  ,use_safetensors=True  FAILS
     #unetX = UNet2DConditionModel.from_pretrained('SG161222/RealVisXL_V5.0',subfolder='unet')  #  ,use_safetensors=True  FAILS
   #  vae = AutoencoderKL.from_pretrained("BeastHF/MyBack_SDXL_Juggernaut_XL_VAE/MyBack_SDXL_Juggernaut_XL_VAE_V10(version_X).safetensors",safety_checker=None).to(torch.bfloat16)
@@ -249,8 +249,8 @@ def generate_30(
     gc.collect()
     global models
     pipe = models[model_choice]
-    if juggernaut == True:
-        pipe.vae=vaeX
+    #if juggernaut == True:
+    #    pipe.vae=vaeX
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator(device='cuda').manual_seed(seed)
     #prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
@@ -311,8 +311,8 @@ def generate_60(
     gc.collect()
     global models
     pipe = models[model_choice]
-    if juggernaut == True:
-        pipe.vae=vaeX
+    #if juggernaut == True:
+    #    pipe.vae=vaeX
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator(device='cuda').manual_seed(seed)
     #prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
@@ -373,8 +373,8 @@ def generate_90(
     gc.collect()
     global models
     pipe = models[model_choice]
-    if juggernaut == True:
-        pipe.vae=vaeX
+    #if juggernaut == True:
+    #    pipe.vae=vaeX
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator(device='cuda').manual_seed(seed)
     #prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)