Skip to content

Commit 4132f7a

Browse files
bug fix
1 parent 24f0423 commit 4132f7a

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

cape_document_qa/training/cape_productionize_model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ def copy_word_vectors(train_config, output_dir):
4242
def copy_elmo_resources(output_dir):
4343
copy2(LM_WEIGHTS, output_dir)
4444
copy2(LM_OPTIONS, output_dir)
45-
copy2(LM_TOKEN_WEIGHTS, output_dir) # possibly remove, not necessary
45+
copy2(LM_VOCAB, output_dir)
46+
#copy2(LM_TOKEN_WEIGHTS, output_dir) # possibly remove, not necessary
4647

4748

4849
def productionize(model_dir, output_dir, train_config=None, convert_from_cudnn=True):

0 commit comments

Comments
 (0)