Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|