Spaces:

Pixelkram
/

webui

Runtime error

Pixelkram commited on Jul 7, 2023

Commit

612d412

1 Parent(s): fb477c0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ repo_id = "stabilityai/stable-diffusion-2"
 scheduler = EulerDiscreteScheduler.from_pretrained(repo_id, subfolder="scheduler", prediction_type="v_prediction")
 pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.float16, revision="fp16", scheduler=scheduler)
 pipe = pipe.to(device)
-pipe.enable_xformers_memory_efficient_attention()
 #If you have duplicated this Space or is running locally, you can remove this snippet
 if "HUGGING_FACE_HUB_TOKEN" in os.environ:

 scheduler = EulerDiscreteScheduler.from_pretrained(repo_id, subfolder="scheduler", prediction_type="v_prediction")
 pipe = DiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch.float16, revision="fp16", scheduler=scheduler)
 pipe = pipe.to(device)
+#pipe.enable_xformers_memory_efficient_attention()
 #If you have duplicated this Space or is running locally, you can remove this snippet
 if "HUGGING_FACE_HUB_TOKEN" in os.environ: