pablo commited on
Commit
7f96cda
·
1 Parent(s): 2e08ffe
Files changed (1) hide show
  1. app.py +1 -2
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)