File size: 646 Bytes
c9ace58
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from sentence_transformers import SentenceTransformer
import torch

EMB_MODEL = "intfloat/multilingual-e5-base"

print(f"Torch version: {torch.__version__}")

try:
    print("Attempt 1: Loading with device='cpu'")
    model = SentenceTransformer(EMB_MODEL, device="cpu", model_kwargs={"low_cpu_mem_usage": False})
    print("Success 1")
except Exception as e:
    print(f"Failed 1: {e}")

try:
    print("Attempt 2: Loading with device='cpu' and low_cpu_mem_usage=False")
    model = SentenceTransformer(EMB_MODEL, device="cpu", model_kwargs={"low_cpu_mem_usage": False})
    print("Success 2")
except Exception as e:
    print(f"Failed 2: {e}")