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)
|