KeyError: 'vit_t'
realcarlos opened this issue · 1 comments
~/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 "remove_anything.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'
@realcarlos Can you please let me know if you found the solution for this issue? I am running into the same issue.