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
|