bitdribble commited on
Commit
4220840
1 Parent(s): e8bab0d

Add SetFit model

Browse files
1_Pooling/config.json CHANGED
@@ -5,5 +5,6 @@
5
  "pooling_mode_max_tokens": false,
6
  "pooling_mode_mean_sqrt_len_tokens": false,
7
  "pooling_mode_weightedmean_tokens": false,
8
- "pooling_mode_lasttoken": false
 
9
  }
 
5
  "pooling_mode_max_tokens": false,
6
  "pooling_mode_mean_sqrt_len_tokens": false,
7
  "pooling_mode_weightedmean_tokens": false,
8
+ "pooling_mode_lasttoken": false,
9
+ "include_prompt": true
10
  }
README.md CHANGED
@@ -161,12 +161,12 @@ preds = model("a fast , funny , highly enjoyable movie . ")
161
 
162
  * The bold row denotes the saved checkpoint.
163
  ### Framework Versions
164
- - Python: 3.11.7
165
  - SetFit: 1.0.3
166
- - Sentence Transformers: 2.3.1
167
  - Transformers: 4.37.2
168
- - PyTorch: 2.2.0+cu121
169
- - Datasets: 2.17.1
170
  - Tokenizers: 0.15.2
171
 
172
  ## Citation
 
161
 
162
  * The bold row denotes the saved checkpoint.
163
  ### Framework Versions
164
+ - Python: 3.12.1
165
  - SetFit: 1.0.3
166
+ - Sentence Transformers: 2.4.0
167
  - Transformers: 4.37.2
168
+ - PyTorch: 2.2.1+cu121
169
+ - Datasets: 2.17.0
170
  - Tokenizers: 0.15.2
171
 
172
  ## Citation
config_sentence_transformers.json CHANGED
@@ -3,5 +3,7 @@
3
  "sentence_transformers": "2.0.0",
4
  "transformers": "4.7.0",
5
  "pytorch": "1.9.0+cu102"
6
- }
 
 
7
  }
 
3
  "sentence_transformers": "2.0.0",
4
  "transformers": "4.7.0",
5
  "pytorch": "1.9.0+cu102"
6
+ },
7
+ "prompts": {},
8
+ "default_prompt_name": null
9
  }