Yaron Koresh commited on
Commit
6a374de
·
verified ·
1 Parent(s): 62c5b0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -17,7 +17,6 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
17
  if torch.cuda.is_available():
18
  torch.cuda.max_memory_allocated(device=device)
19
  pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
20
- pipe.enable_xformers_memory_efficient_attention()
21
  pipe = pipe.to(device)
22
  else:
23
  pipe = DiffusionPipeline.from_pretrained(model_id, use_safetensors=True)
 
17
  if torch.cuda.is_available():
18
  torch.cuda.max_memory_allocated(device=device)
19
  pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
 
20
  pipe = pipe.to(device)
21
  else:
22
  pipe = DiffusionPipeline.from_pretrained(model_id, use_safetensors=True)