Spaces:
Running
Running
nssharmaofficial
commited on
Commit
•
e8a30b2
1
Parent(s):
3f0aa35
Config old model
Browse files- source/config.py +3 -3
source/config.py
CHANGED
@@ -16,7 +16,7 @@ class Config(object):
|
|
16 |
self.WORD_EMB_DIM = 256
|
17 |
self.HIDDEN_DIM = 512
|
18 |
|
19 |
-
self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-
|
20 |
-
self.ENCODER_WEIGHT_FILE = 'source/weights/encoder-32B-512H-1L-
|
21 |
-
self.DECODER_WEIGHT_FILE = 'source/weights/decoder-32B-512H-1L-
|
22 |
|
|
|
16 |
self.WORD_EMB_DIM = 256
|
17 |
self.HIDDEN_DIM = 512
|
18 |
|
19 |
+
self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e5.pt'
|
20 |
+
self.ENCODER_WEIGHT_FILE = 'source/weights/encoder-32B-512H-1L-e5.pt'
|
21 |
+
self.DECODER_WEIGHT_FILE = 'source/weights/decoder-32B-512H-1L-e5.pt'
|
22 |
|