geekyutao / Inpaint-Anything

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

KeyError: 'vit_t' #110

Open realcarlos opened 11 months ago

realcarlos commented 11 months ago

~/Inpaint-Anything$ bash ./script/remove_anything.sh 2023-10-18 22:10:34.265534: I tensorflow/core/platform/cpu_feature_guard.cc:182] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations. To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags. 2023-10-18 22:10:34.850371: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT Detectron v2 is not installed Traceback (most recent call last): File "removeanything.py", line 86, in masks, , _ = predict_masks_with_sam( File "/home/xxx/Inpaint-Anything/sam_segment.py", line 24, in predict_masks_with_sam sam = sam_model_registrymodel_type KeyError: 'vit_t'

alireza18878 commented 9 months ago

@realcarlos Can you please let me know if you found the solution for this issue? I am running into the same issue.

nisel1999 commented 1 week ago

You can solve this problem by removing “segment_anything” package.