Manjushri commited on
Commit
b6b2770
·
verified ·
1 Parent(s): 2555f1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ torch.cuda.max_memory_allocated(device=device)
13
  torch.cuda.empty_cache()
14
  torch.cuda.max_memory_allocated(device=device)
15
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium", torch_dtype=torch.bfloat16).to(device)
16
- pipe.enable_xformers_memory_efficient_attention()
17
  #pipe.enable_model_cpu_offload()
18
  torch.cuda.empty_cache()
19
 
 
13
  torch.cuda.empty_cache()
14
  torch.cuda.max_memory_allocated(device=device)
15
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium", torch_dtype=torch.bfloat16).to(device)
16
+ #pipe.enable_xformers_memory_efficient_attention()
17
  #pipe.enable_model_cpu_offload()
18
  torch.cuda.empty_cache()
19