Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
jichao commited on
Commit ·
7064790
1
Parent(s): 48e1f93
add new model
Browse files- .claude/settings.local.json +7 -0
- .gitattributes +2 -0
- app.py +7 -1
- models/dino-vits-mae-100epoch-1217-1220-e50.pth +3 -0
.claude/settings.local.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"permissions": {
|
| 3 |
+
"allow": [
|
| 4 |
+
"Bash(dir:*)"
|
| 5 |
+
]
|
| 6 |
+
}
|
| 7 |
+
}
|
.gitattributes
CHANGED
|
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
|
| 37 |
+
.claude/
|
app.py
CHANGED
|
@@ -7,7 +7,7 @@ import numpy as np
|
|
| 7 |
import os
|
| 8 |
|
| 9 |
# --- Model Configuration ---
|
| 10 |
-
DEFAULT_MODEL_NAME = "
|
| 11 |
MODEL_CONFIGS = {
|
| 12 |
"mars-ctx-vitb-0217": {
|
| 13 |
"path": "models/0217-checkpoint-300.pth",
|
|
@@ -28,6 +28,12 @@ MODEL_CONFIGS = {
|
|
| 28 |
"in_chans": 1,
|
| 29 |
"description": "ViT-Small/16 (Grayscale Input)"
|
| 30 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
}
|
| 32 |
|
| 33 |
# Global dictionary to store loaded models
|
|
|
|
| 7 |
import os
|
| 8 |
|
| 9 |
# --- Model Configuration ---
|
| 10 |
+
DEFAULT_MODEL_NAME = "dino-vits-mae-100epoch-1217-1220-e50"
|
| 11 |
MODEL_CONFIGS = {
|
| 12 |
"mars-ctx-vitb-0217": {
|
| 13 |
"path": "models/0217-checkpoint-300.pth",
|
|
|
|
| 28 |
"in_chans": 1,
|
| 29 |
"description": "ViT-Small/16 (Grayscale Input)"
|
| 30 |
},
|
| 31 |
+
"dino-vits-mae-100epoch-1217-1220-e50": {
|
| 32 |
+
"path": "models/dino-vits-mae-100epoch-1217-1220-e50.pth",
|
| 33 |
+
"timm_id": "vit_small_patch16_224",
|
| 34 |
+
"in_chans": 1,
|
| 35 |
+
"description": "ViT-Small/16 DINO+MAE (Grayscale Input)"
|
| 36 |
+
},
|
| 37 |
}
|
| 38 |
|
| 39 |
# Global dictionary to store loaded models
|
models/dino-vits-mae-100epoch-1217-1220-e50.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5a07ee0c6a6787a4cd6ef52b43a4bcfa5efa1499c7ca15a3c03c0e49c11819c5
|
| 3 |
+
size 85933931
|