Upload luciddreamer.py with huggingface_hub

luciddreamer.py

@@ -69,7 +69,7 @@ class LucidDreamer:
         self.background = torch.tensor(bg_color, dtype=torch.float32, device='cuda')
         
         self.rgb_model = StableDiffusionInpaintPipeline.from_pretrained(
-            'runwayml/stable-diffusion-inpainting', revision='fp16', torch_dtype=torch.float16).to('cuda')
+            'stablediffusion/SD1-5', revision='fp16', torch_dtype=torch.float16).to('cuda')
         self.d_model = torch.hub.load('./ZoeDepth', 'ZoeD_N', source='local', pretrained=True).to('cuda')
         self.controlnet = None
         self.lama = None
@@ -84,7 +84,8 @@ class LucidDreamer:
             self.controlnet = None
             self.lama = None
             self.rgb_model = StableDiffusionInpaintPipeline.from_pretrained(
-                'runwayml/stable-diffusion-inpainting',
+                #  'runwayml/stable-diffusion-inpainting',
+                'stablediffusion/SD1-5',
                 revision='fp16',
                 torch_dtype=torch.float16,
                 safety_checker=None,
@@ -528,4 +529,4 @@ class LucidDreamer:
                 })
 
         progress(1, desc='Baking Gaussians...')
-        return traindata
+        return traindata