ashishraics commited on
Commit
dc3d11f
1 Parent(s): a388d0d

updated config

Browse files
Files changed (1) hide show
  1. config.yaml +14 -0
config.yaml CHANGED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ SENTIMENT_CLF:
2
+ sent_chkpt: 'distilbert-base-uncased-finetuned-sst-2-english'
3
+ sent_mdl_dir: 'sentiment_model_dir'
4
+ sent_onnx_mdl_dir: 'sent_clf_onnx_dir'
5
+ sent_onnx_mdl_name: 'sentiment_classifier_onnx.onnx'
6
+ sent_onnx_quant_mdl_name: 'sentiment_classifier_onnx_quant.onnx'
7
+
8
+ ZEROSHOT_CLF:
9
+ zs_chkpt: 'valhalla/distilbart-mnli-12-1'
10
+ zs_mdl_dir: 'zs_model_dir'
11
+ zs_onnx_mdl_dir: 'zs_onnx_dir'
12
+ zs_onnx_mdl_name: 'model.onnx'
13
+ zs_onnx_quant_mdl_name: 'model_quant.onnx'
14
+