PhyscalX commited on
Commit
764cc00
1 Parent(s): 437fe89

Add ViT-H v1.1 model

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. models/tap_vit_h_v1_1.pkl +3 -0
app.py CHANGED
@@ -32,7 +32,7 @@ def parse_args():
32
  """Parse arguments."""
33
  parser = argparse.ArgumentParser(description="Launch gradio application")
34
  parser.add_argument("--model-type", type=str, default="tap_vit_l")
35
- parser.add_argument("--checkpoint", type=str, default="models/tap_vit_l_v1_1.pkl")
36
  parser.add_argument("--concept", type=str, default="concepts/merged_2560.pkl")
37
  parser.add_argument("--device", nargs="+", type=int, default=[0], help="Index of devices")
38
  return parser.parse_args()
 
32
  """Parse arguments."""
33
  parser = argparse.ArgumentParser(description="Launch gradio application")
34
  parser.add_argument("--model-type", type=str, default="tap_vit_l")
35
+ parser.add_argument("--checkpoint", type=str, default="models/tap_vit_h_v1_1.pkl")
36
  parser.add_argument("--concept", type=str, default="concepts/merged_2560.pkl")
37
  parser.add_argument("--device", nargs="+", type=int, default=[0], help="Index of devices")
38
  return parser.parse_args()
models/tap_vit_h_v1_1.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d3caabfc826ad52f2c34f4fe6f575b70189d3effe837da9cf3264d080c40227
3
+ size 1488947633