Update app.py

app.py

@@ -75,7 +75,7 @@ def Loading(model_name, is_xl, is_cuda):
 		text2img = pipeline_class.from_pretrained(model_name, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to(device)
 	else:
 		text2img = pipeline_class.from_pretrained(model_name, use_safetensors=True).to(device)
-	if is_xl:
+	if is_cuda:
 		text2img.enable_xformers_memory_efficient_attention()
 		text2img.vae.enable_xformers_memory_efficient_attention()
 	text2img.safety_checker = None
@@ -83,7 +83,7 @@ def Loading(model_name, is_xl, is_cuda):
 	return model_name
 with gr.Blocks() as demo:
 	with gr.Row():
-		model_name = gr.Textbox(value="nota-ai/bk-sdm-tiny-2m", label=translations[language]['model_name'])
+		model_name = gr.Textbox(value="John6666/pony-realism-v21main-sdxl", label=translations[language]['model_name'])
 		is_xl = gr.Checkbox(label="SDXL")
 		is_cuda = gr.Checkbox(label="cuda", value=torch.cuda.is_available())
 		loading = gr.Button(translations[language]['loading'])