Muhammadreza commited on
Commit
f2d86fb
1 Parent(s): 0c1adc7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -10,6 +10,9 @@ import cv2
10
  pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams", torch_dtype=torch.float16).to("cuda")
11
  pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True)
12
 
 
 
 
13
  torch.cuda.empty_cache()
14
 
15
  @spaces.GPU
 
10
  pipe = DiffusionPipeline.from_pretrained("mann-e/Mann-E_Dreams", torch_dtype=torch.float16).to("cuda")
11
  pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, use_karras_sigmas=True)
12
 
13
+ pipe.enable_xformers_memory_efficient_attention()
14
+ pipe.enable_vae_slicing()
15
+
16
  torch.cuda.empty_cache()
17
 
18
  @spaces.GPU