diff --git a/ucsinfer/__main__.py b/ucsinfer/__main__.py
index 9c98514..9ef05b2 100644
--- a/ucsinfer/__main__.py
+++ b/ucsinfer/__main__.py
@@ -246,7 +246,10 @@ def evaluate(dataset, offset, limit, model, no_foley):
 if __name__ == '__main__':
     os.environ['TOKENIZERS_PARALLELISM'] = 'false'
 
+    # sentence_transformers generates an error in PyTorch upon loading
     import warnings
-    warnings.simplefilter(action='ignore', category=FutureWarning)
+    warnings.filterwarnings(action='ignore', module='torch', 
+                            category=FutureWarning, 
+                            message=r"`encoder_attention_mask` is deprecated.*")
 
     ucsinfer()