Made the warnings filter more specific
This commit is contained in:
@@ -246,7 +246,10 @@ def evaluate(dataset, offset, limit, model, no_foley):
|
|||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
os.environ['TOKENIZERS_PARALLELISM'] = 'false'
|
os.environ['TOKENIZERS_PARALLELISM'] = 'false'
|
||||||
|
|
||||||
|
# sentence_transformers generates an error in PyTorch upon loading
|
||||||
import warnings
|
import warnings
|
||||||
warnings.simplefilter(action='ignore', category=FutureWarning)
|
warnings.filterwarnings(action='ignore', module='torch',
|
||||||
|
category=FutureWarning,
|
||||||
|
message=r"`encoder_attention_mask` is deprecated.*")
|
||||||
|
|
||||||
ucsinfer()
|
ucsinfer()
|
||||||
|
Reference in New Issue
Block a user