Fix HF image predictor

2024-08-12 23:41:41 +00:00
parent dce7b5446f
commit 1191677e1e
3 changed files with 9 additions and 4 deletions
--- a/sam2/build_sam.py
+++ b/sam2/build_sam.py
@@ -19,6 +19,7 @@ def build_sam2(
    mode="eval",
    hydra_overrides_extra=[],
    apply_postprocessing=True,
+    **kwargs,
 ):

    if apply_postprocessing:
@@ -47,6 +48,7 @@ def build_sam2_video_predictor(
    mode="eval",
    hydra_overrides_extra=[],
    apply_postprocessing=True,
+    **kwargs,
 ):
    hydra_overrides = [
        "++model._target_=sam2.sam2_video_predictor.SAM2VideoPredictor",