ford442 commited on
Commit
5dd3470
·
verified ·
1 Parent(s): 4b9e6c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -108,8 +108,6 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str
108
  return p.replace("{prompt}", positive), n + negative
109
 
110
  def load_and_prepare_model(model_id):
111
- model_dtypes = {"ford442/RealVisXL_V5.0_BF16": torch.bfloat16,}
112
- dtype = model_dtypes.get(model_id, torch.bfloat16) # Default to float32 if not found
113
  #vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None)
114
  vaeX = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None,use_safetensors=False)
115
  #vae = AutoencoderKL.from_pretrained('cross-attention/asymmetric-autoencoder-kl-x-2',use_safetensors=False)
 
108
  return p.replace("{prompt}", positive), n + negative
109
 
110
  def load_and_prepare_model(model_id):
 
 
111
  #vae = AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", safety_checker=None)
112
  vaeX = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None,use_safetensors=False)
113
  #vae = AutoencoderKL.from_pretrained('cross-attention/asymmetric-autoencoder-kl-x-2',use_safetensors=False)