jke94 commited on
Commit
fcba6a3
1 Parent(s): b120c1f

Remove line about xformers

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -40,11 +40,6 @@ def predict(image_input, prompt, negative_prompt, sd_models_dropdown):
40
 
41
  # speed up diffusion process with faster scheduler and memory optimization
42
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
43
-
44
- # remove following line if xformers is not installed
45
- pipe.enable_xformers_memory_efficient_attention()
46
-
47
- pipe.enable_model_cpu_offload()
48
 
49
  # prompt = "a photo of an astronaut riding a horse on mars"
50
  # negative_prompt=""
 
40
 
41
  # speed up diffusion process with faster scheduler and memory optimization
42
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 
 
 
 
 
43
 
44
  # prompt = "a photo of an astronaut riding a horse on mars"
45
  # negative_prompt=""