Spaces:
Running
on
Zero
Running
on
Zero
disable xformers for demo
Browse files- 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 |
-
|
173 |
-
|
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
|