czl commited on
Commit
2b1fdf1
1 Parent(s): 067a15e

disable xformers for demo

Browse files
Files changed (1) hide show
  1. tools/synth.py +3 -3
tools/synth.py CHANGED
@@ -168,9 +168,9 @@ def pipe_img(
168
  cache_interval=cache_interval, cache_branch_id=cache_branch_id
169
  ) # lower is faster but lower quality
170
  helper.enable()
171
- if torch.cuda.is_available():
172
- pipe.to("cuda")
173
- pipe.enable_xformers_memory_efficient_attention()
174
  if use_torchcompile:
175
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
176
  return pipe
 
168
  cache_interval=cache_interval, cache_branch_id=cache_branch_id
169
  ) # lower is faster but lower quality
170
  helper.enable()
171
+ # if torch.cuda.is_available():
172
+ # pipe.to("cuda")
173
+ # pipe.enable_xformers_memory_efficient_attention()
174
  if use_torchcompile:
175
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
176
  return pipe