Spaces:
Sleeping
Sleeping
Add ViT-H v1.1 model
Browse files- app.py +1 -1
- 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/
|
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
|