Spaces:

lbw18601752667
/

IDMR-demo

Running

liubangwei commited on Mar 25

Commit

45123ea

1 Parent(s): 3ee7efa

fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,8 +22,8 @@ def load_model():
     global IMAGE_TOKEN
     model_args = ModelArguments(
-        model_name="/fs-computility/ai-shen/kilab-shared/liubangwei/ckpt/my_hf/IDMR-2B",
-        # model_name="lbw18601752667/IDMR-2B",
         model_backbone="internvl_2_5",
     )

     global IMAGE_TOKEN
     model_args = ModelArguments(
+        # model_name="/fs-computility/ai-shen/kilab-shared/liubangwei/ckpt/my_hf/IDMR-2B",
+        model_name="lbw18601752667/IDMR-2B",
         model_backbone="internvl_2_5",
     )