back compat predict.py

2018-12-10 19:41:07 +00:00 · 2018-12-10 19:41:07 +00:00 · 1bd922b5bc
parent 9a4493b655
commit 1bd922b5bc
1 changed files with 4 additions and 0 deletions
--- a/predict.py
+++ b/predict.py
@ -213,6 +213,10 @@ def get_args():
                setattr(args, r,  config[r])
            elif 'cove' in r:
                setattr(args, r, False)
+            elif 'elmo' in r:
+                setattr(args, r, False)
+            elif 'glove_and_char' in r:
+                setattr(args, r, True)
            else:
                setattr(args, r, None)
        args.dropout_ratio = 0.0