diff --git a/combo/main.py b/combo/main.py index e2e29d437ad31e24c69fe6c95c60d29a8491e085..918bb9407083e9babe3c53ee1671b412fba5fe74 100644 --- a/combo/main.py +++ b/combo/main.py @@ -143,6 +143,8 @@ def run(_): file.writelines(predictor.predict_instance_as_tree(tree).serialize()) else: use_dataset_reader = ".conllu" in FLAGS.input_file.lower() + if FLAGS.silent: + logging.getLogger("allennlp.common.params").disabled = True manager = allen_predict._PredictManager( _get_predictor(), FLAGS.input_file, diff --git a/setup.py b/setup.py index ee68a63bd843b576b334b68af5a6b20c4f4b09c1..328d05d4e97bbaf767a9a973abd665124610e43d 100644 --- a/setup.py +++ b/setup.py @@ -10,8 +10,8 @@ REQUIREMENTS = [ 'requests==2.23.0', 'overrides==3.0.0', 'tensorboard==2.1.0', - 'torch==1.5.0', - 'torchvision==0.6.0', + 'torch>=1.5.0', + 'torchvision>=0.6.0', 'tqdm==4.43.0', 'transformers==2.9.1', 'urllib3==1.24.2',