Spaces:
Sleeping
Sleeping
nssharmaofficial
commited on
Commit
•
9f7e6ed
1
Parent(s):
9a90e40
Fix weights path
Browse files- source/config.py +3 -3
source/config.py
CHANGED
@@ -16,7 +16,7 @@ class Config(object):
|
|
16 |
self.WORD_EMB_DIM = 5121
|
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 = 5121
|
17 |
self.HIDDEN_DIM = 512
|
18 |
|
19 |
+
self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e10.pt'
|
20 |
+
self.ENCODER_WEIGHT_FILE = 'source/weights/encoder-32B-512H-1L-e10.pt'
|
21 |
+
self.DECODER_WEIGHT_FILE = 'source/weights/decoder-32B-512H-1L-e10.pt'
|
22 |
|