sattwik21 commited on
Commit
fca8613
·
verified ·
1 Parent(s): bc1ab8a

Upload GestrJEPAForClassification

Browse files
Files changed (2) hide show
  1. config.json +4 -0
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,8 +1,12 @@
1
  {
 
 
 
2
  "auto_map": {
3
  "AutoConfig": "modeling_gestr_jepa.GestrJEPAConfig",
4
  "AutoModelForSequenceClassification": "modeling_gestr_jepa.GestrJEPAForClassification"
5
  },
 
6
  "embed_dim": 64,
7
  "hidden_dim": 256,
8
  "id2label": {
 
1
  {
2
+ "architectures": [
3
+ "GestrJEPAForClassification"
4
+ ],
5
  "auto_map": {
6
  "AutoConfig": "modeling_gestr_jepa.GestrJEPAConfig",
7
  "AutoModelForSequenceClassification": "modeling_gestr_jepa.GestrJEPAForClassification"
8
  },
9
+ "dtype": "float32",
10
  "embed_dim": 64,
11
  "hidden_dim": 256,
12
  "id2label": {
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1c4872b9afc1242e46d59e4aa939e2b50c4e4b5b078f2cea04fdd75e7c5ae12b
3
  size 354464
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a3c87c94196877ce1e5c34f4dcaea63bf4d18db8b7558598cf40c8e59705558
3
  size 354464