bitdribble
commited on
Commit
•
4220840
1
Parent(s):
e8bab0d
Add SetFit model
Browse files- 1_Pooling/config.json +2 -1
- README.md +4 -4
- config_sentence_transformers.json +3 -1
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.
|
165 |
- SetFit: 1.0.3
|
166 |
-
- Sentence Transformers: 2.
|
167 |
- Transformers: 4.37.2
|
168 |
-
- PyTorch: 2.2.
|
169 |
-
- Datasets: 2.17.
|
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 |
}
|