geekyutao / Inpaint-Anything

Inpaint anything using Segment Anything and inpainting models.
Apache License 2.0
6.4k stars 525 forks source link

remove anything inference issue #146

Open bekhzod-olimov opened 6 months ago

bekhzod-olimov commented 6 months ago

Hi guys! I am trying to run inference code provided in the README documentation. I have installed all the requirement libraries, downloaded necessary checkpoints and placed them as it is said in the documentation. I have 4 NVIDIA A100 GPUs (CUDA 12.0) with 40GB each and have enough spaces in all of them. I have used the same script as in the README as follows:

python remove_anything.py --input_img ./example/remove-anything/dog.jpg --coords_type key_in --point_coords 200 450 --point_labels 1 --dilate_kernel_size 15 --output_dir ./results --sam_model_type "vit_h" --sam_ckpt ./pretrained_models/sam_vit_h_4b8939.pth --lama_config ./lama/configs/prediction/default.yaml --lama_ckpt ./pretrained_models/big-lama

After running the code I have

Aborted (core dumped)

error.

image

What can be the reason for it? And how can I solve it?