jaisidhsingh commited on
Commit
58c612d
·
1 Parent(s): fa56924

Upload model_config.py

Browse files
Files changed (1) hide show
  1. configs/model_config.py +11 -0
configs/model_config.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from types import SimpleNamespace
2
+
3
+
4
+ cfg = SimpleNamespace(**{})
5
+
6
+ # sentence embedding model configs
7
+ cfg.sent_model_name = "sentence-transformers/all-MiniLM-L6-v2"
8
+ cfg.sent_model_seq_limit = 256
9
+
10
+ # summarization model configs
11
+