KameliaZaman commited on
Commit
5abbe0c
1 Parent(s): 7727851

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +20 -13
config.json CHANGED
@@ -1,15 +1,22 @@
1
  {
2
- "architectures": [
3
- "Seq2SeqTransformer"
4
- ],
5
- "dim_feedforward": 512,
6
- "dropout": 0.1,
7
- "emb_size": 512,
8
- "nhead": 8,
9
- "num_decoder_layers": 3,
10
- "num_encoder_layers": 3,
11
- "src_vocab_size": 19214,
12
- "tgt_vocab_size": 10837,
13
- "torch_dtype": "float32",
14
- "transformers_version": "4.38.2"
 
 
 
 
 
 
 
15
  }
 
1
  {
2
+ "languages": {
3
+ "fr": "French",
4
+ "en": "English"
5
+ },
6
+ "models": [
7
+ {
8
+ "src": "fr",
9
+ "tgt": "en",
10
+ "model_type": "ctranslator2",
11
+ "hugging_face_repo_id": "KameliaZaman/French-to-English-Translation" // repo id to download the model (optional)",
12
+ "model_path": "French-to-English-Translation", //model directory name under models
13
+ "src_sentencepiece_model": "french_to_english_translator.h5",
14
+ "tgt_sentencepiece_model": "french_to_english_translator.h5",
15
+ "sentence_split": "nltk",
16
+ "pipeline": {
17
+ "sentencepiece": true,
18
+ "translate": true
19
+ }
20
+ }
21
+ ]
22
  }