Skip to content

Commit 658cd3f

Browse files
authored
fix(model): Fix inpainting model missing. (#69)
1 parent 47068d7 commit 658cd3f

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

embodied_gen/utils/monkey_patches.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,8 @@ def patched_lama_inpaint_init(self):
211211
def patched_sd_inpaint_init(self, subset_name=None):
212212
super(SDFTInpainter, self).__init__()
213213
pipe = StableDiffusionInpaintPipeline.from_pretrained(
214-
"stabilityai/stable-diffusion-2-inpainting",
214+
# "stabilityai/stable-diffusion-2-inpainting",
215+
"sd2-community/stable-diffusion-2-inpainting",
215216
torch_dtype=torch.float16,
216217
).to("cuda")
217218
pipe.enable_model_cpu_offload()

0 commit comments

Comments
 (0)