Spaces:
Runtime error
Runtime error
pablo
commited on
Commit
·
7f96cda
1
Parent(s):
2e08ffe
nmew
Browse files
app.py
CHANGED
|
@@ -13,11 +13,10 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 13 |
|
| 14 |
# Inpainting pipeline
|
| 15 |
|
| 16 |
-
|
| 17 |
unet = UNet2DConditionModel.from_pretrained("pablodawson/ldm3d-inpainting", cache_dir="cache", subfolder="unet", in_channels=9, low_cpu_mem_usage=False, ignore_mismatched_sizes=True)
|
| 18 |
pipe = StableDiffusionLDM3DInpaintPipeline.from_pretrained("Intel/ldm3d-4c", cache_dir="cache" ).to(device)
|
| 19 |
|
| 20 |
-
'''
|
| 21 |
# Depth estimation
|
| 22 |
model_type = "DPT_Large" # MiDaS v3 - Large (highest accuracy, slowest inference speed)
|
| 23 |
#model_type = "DPT_Hybrid" # MiDaS v3 - Hybrid (medium accuracy, medium inference speed)
|
|
|
|
| 13 |
|
| 14 |
# Inpainting pipeline
|
| 15 |
|
| 16 |
+
|
| 17 |
unet = UNet2DConditionModel.from_pretrained("pablodawson/ldm3d-inpainting", cache_dir="cache", subfolder="unet", in_channels=9, low_cpu_mem_usage=False, ignore_mismatched_sizes=True)
|
| 18 |
pipe = StableDiffusionLDM3DInpaintPipeline.from_pretrained("Intel/ldm3d-4c", cache_dir="cache" ).to(device)
|
| 19 |
|
|
|
|
| 20 |
# Depth estimation
|
| 21 |
model_type = "DPT_Large" # MiDaS v3 - Large (highest accuracy, slowest inference speed)
|
| 22 |
#model_type = "DPT_Hybrid" # MiDaS v3 - Hybrid (medium accuracy, medium inference speed)
|