Mohamed11 commited on
Commit
403869b
1 Parent(s): 33a11ec

Add SetFit model

Browse files
Files changed (4) hide show
  1. README.md +3 -3
  2. config.json +1 -1
  3. config_setfit.json +2 -2
  4. model_head.pkl +1 -1
README.md CHANGED
@@ -178,10 +178,10 @@ preds = model("President Blasts Firing Notre Dame's outgoing president criticize
178
  - Python: 3.10.13
179
  - SetFit: 1.0.3
180
  - Sentence Transformers: 2.7.0
181
- - Transformers: 4.39.3
182
- - PyTorch: 2.1.2
183
  - Datasets: 2.18.0
184
- - Tokenizers: 0.15.2
185
 
186
  ## Citation
187
 
 
178
  - Python: 3.10.13
179
  - SetFit: 1.0.3
180
  - Sentence Transformers: 2.7.0
181
+ - Transformers: 4.40.2
182
+ - PyTorch: 2.3.0+cu121
183
  - Datasets: 2.18.0
184
+ - Tokenizers: 0.19.1
185
 
186
  ## Citation
187
 
config.json CHANGED
@@ -19,7 +19,7 @@
19
  "pad_token_id": 0,
20
  "position_embedding_type": "absolute",
21
  "torch_dtype": "float32",
22
- "transformers_version": "4.39.3",
23
  "type_vocab_size": 2,
24
  "use_cache": true,
25
  "vocab_size": 30522
 
19
  "pad_token_id": 0,
20
  "position_embedding_type": "absolute",
21
  "torch_dtype": "float32",
22
+ "transformers_version": "4.40.2",
23
  "type_vocab_size": 2,
24
  "use_cache": true,
25
  "vocab_size": 30522
config_setfit.json CHANGED
@@ -1,4 +1,4 @@
1
  {
2
- "labels": null,
3
- "normalize_embeddings": false
4
  }
 
1
  {
2
+ "normalize_embeddings": false,
3
+ "labels": null
4
  }
model_head.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c2faf741199c95bbb8ac74ad5bcb6cf7fc3335875708ed17bdac05cf69274096
3
  size 13191
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34815a23dc91a292e9dc134c97d272672aea1c4264f89cf86d66cb21631c71f1
3
  size 13191