ford442 commited on
Commit
eb7b68b
·
verified ·
1 Parent(s): 6ebd597

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -91,7 +91,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
91
  def load_and_prepare_model(model_id):
92
  model_dtypes = {"ford442/RealVisXL_V5.0_BF16": torch.bfloat16,}
93
  dtype = model_dtypes.get(model_id, torch.bfloat16) # Default to float32 if not found
94
- vaeX = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None, use_safetensors=False)
95
  #vaeX = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None, use_safetensors=False)
96
  #vae = AutoencoderKL.from_pretrained('cross-attention/asymmetric-autoencoder-kl-x-2',use_safetensors=False)
97
  #vae = AutoencoderKL.from_single_file('https://huggingface.co/ford442/sdxl-vae-bf16/mySLR/myslrVAE_v10.safetensors')
@@ -214,7 +214,7 @@ def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp):
214
  f.write(f"SPACE SETUP: \n")
215
  f.write(f"Use Model Dtype: no \n")
216
  f.write(f"Model Scheduler: Euler_a all_custom before cuda \n")
217
- f.write(f"Model VAE: sdxl-vae-bf16 safetensor=false before cuda then attn_proc / scale factor 8 \n")
218
  f.write(f"Model UNET: default ford442/RealVisXL_V5.0_BF16 \n")
219
  f.write(f"Model HiDiffusion OFF \n")
220
  f.write(f"Model do_resize ON \n")
 
91
  def load_and_prepare_model(model_id):
92
  model_dtypes = {"ford442/RealVisXL_V5.0_BF16": torch.bfloat16,}
93
  dtype = model_dtypes.get(model_id, torch.bfloat16) # Default to float32 if not found
94
+ vaeX = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None)
95
  #vaeX = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None, use_safetensors=False)
96
  #vae = AutoencoderKL.from_pretrained('cross-attention/asymmetric-autoencoder-kl-x-2',use_safetensors=False)
97
  #vae = AutoencoderKL.from_single_file('https://huggingface.co/ford442/sdxl-vae-bf16/mySLR/myslrVAE_v10.safetensors')
 
214
  f.write(f"SPACE SETUP: \n")
215
  f.write(f"Use Model Dtype: no \n")
216
  f.write(f"Model Scheduler: Euler_a all_custom before cuda \n")
217
+ f.write(f"Model VAE: sdxl-vae-bf16 before cuda then attn_proc / scale factor 8 \n")
218
  f.write(f"Model UNET: default ford442/RealVisXL_V5.0_BF16 \n")
219
  f.write(f"Model HiDiffusion OFF \n")
220
  f.write(f"Model do_resize ON \n")