ldm3d

Running on T4

estellea commited on Jul 9, 2023

Commit

657f55f

•

1 Parent(s): 71d12ce

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,8 @@ pipe = StableDiffusionLDM3DPipeline.from_pretrained(
     # , safety_checker=None
 )
 pipe.to(device)
-pipe.enable_xformers_memory_efficient_attention()
 pipe.enable_model_cpu_offload()

     # , safety_checker=None
 )
 pipe.to(device)
+if device == "cuda":
+    pipe.enable_xformers_memory_efficient_attention()
 pipe.enable_model_cpu_offload()