Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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"):
|