Merge branch 'master' into feat/efficient_loading

UKPLab · Dec 12, 2023 · 033bf6d · 033bf6d
2 parents a00482f + 944cac0
commit 033bf6d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sentence_transformers/SentenceTransformer.py b/sentence_transformers/SentenceTransformer.py
@@ -4,8 +4,12 @@
 import shutil
 import stat
 from collections import OrderedDict
-from typing import List, Dict, Tuple, Iterable, Type, Union, Callable, Optional, Literal
 import warnings
+from typing import List, Dict, Tuple, Iterable, Type, Union, Callable, Optional
+try:
+    from typing import Literal
+except ImportError:
+    from typing_extensions import Literal
 import numpy as np
 from numpy import ndarray
 import transformers