From d5bb399a42bb3db27dd73fc957b3983befefcca5 Mon Sep 17 00:00:00 2001 From: Heng Yang Date: Thu, 31 Aug 2023 00:28:18 +0100 Subject: [PATCH] Update multilingual_universal_sentence_encoder.py --- .../multilingual_universal_sentence_encoder.py | 16 +++------------- 1 file changed, 3 insertions(+), 13 deletions(-) diff --git a/textattack/constraints/semantics/sentence_encoders/universal_sentence_encoder/multilingual_universal_sentence_encoder.py b/textattack/constraints/semantics/sentence_encoders/universal_sentence_encoder/multilingual_universal_sentence_encoder.py index 06dd8478..54905f54 100644 --- a/textattack/constraints/semantics/sentence_encoders/universal_sentence_encoder/multilingual_universal_sentence_encoder.py +++ b/textattack/constraints/semantics/sentence_encoders/universal_sentence_encoder/multilingual_universal_sentence_encoder.py @@ -20,19 +20,13 @@ def __init__(self, threshold=0.8, large=False, metric="angular", **kwargs): tensorflow_text._load() if large: tfhub_url = "https://tfhub.dev/google/universal-sentence-encoder-multilingual-large/3" - mirror_tfhub_url = "https://hub.tensorflow.google.cn/google/universal-sentence-encoder-multilingual-large/3" else: tfhub_url = "https://https://tfhub.dev/google/universal-sentence-encoder-multilingual/3" - mirror_tfhub_url = "https://hub.tensorflow.google.cn/google/universal-sentence-encoder-multilingual/3" # TODO add QA SET. Details at: https://hub.tensorflow.google.cn/google/universal-sentence-encoder-multilingual-qa/3 self._tfhub_url = tfhub_url - self.mirror_tfhub_url = mirror_tfhub_url - try: - self.model = hub.load(self._tfhub_url) - except Exception as e: - print('Error loading model from tfhub, trying mirror url') - self.model = hub.load(self.mirror_tfhub_url) + self.model = hub.load(self._tfhub_url) + def encode(self, sentences): return self.model(sentences).numpy() @@ -44,8 +38,4 @@ def __getstate__(self): def __setstate__(self, state): self.__dict__ = state - try: - self.model = hub.load(self._tfhub_url) - except Exception as e: - print('Error loading model from tfhub, trying mirror url') - self.model = hub.load(self.mirror_tfhub_url) + self.model = hub.load(self._tfhub_url)