radames commited on
Commit
9bf6ab3
·
verified ·
1 Parent(s): b2ab83e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ SAFETY_CHECKER = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-di
19
 
20
  UPSCALER = DiffusionPipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16)
21
  UPSCALER.to("cuda")
22
- UPSCALER.enable_xformers_memory_efficient_attention()
23
 
24
 
25
  class Model:
 
19
 
20
  UPSCALER = DiffusionPipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16)
21
  UPSCALER.to("cuda")
22
+ # UPSCALER.enable_xformers_memory_efficient_attention()
23
 
24
 
25
  class Model: