estellea commited on
Commit
942501f
1 Parent(s): b796e0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ pipe = StableDiffusionLDM3DPipeline.from_pretrained(
19
  pipe.to(device)
20
  if device == "cuda":
21
  pipe.enable_xformers_memory_efficient_attention()
22
- pipe.enable_model_cpu_offload()
23
 
24
 
25
  def get_iframe(rgb_path: str, depth_path: str, viewer_mode: str = "6DOF"):
 
19
  pipe.to(device)
20
  if device == "cuda":
21
  pipe.enable_xformers_memory_efficient_attention()
22
+ pipe.enable_model_cpu_offload()
23
 
24
 
25
  def get_iframe(rgb_path: str, depth_path: str, viewer_mode: str = "6DOF"):