Spaces:

Eugeoter
/

ControlNeXt

Running on Zero

Eugeoter commited on Aug 19, 2024

Commit

fc2d564

1 Parent(s): f92f1ef

test

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,7 +39,7 @@ def ui():
         device=device,
         hf_cache_dir=CACHE_DIR,
         use_safetensors=True,
-        enable_xformers_memory_efficient_attention=True,
     )
     preprocessors = ['canny']

         device=device,
         hf_cache_dir=CACHE_DIR,
         use_safetensors=True,
+        enable_xformers_memory_efficient_attention=torch.cuda.is_available(),
     )
     preprocessors = ['canny']