1bincooo commited on
Commit
6d68d79
·
verified ·
1 Parent(s): 3cf2d78

Update localembedding.py

Browse files
Files changed (1) hide show
  1. localembedding.py +1 -1
localembedding.py CHANGED
@@ -32,7 +32,7 @@ if torch.cuda.is_available():
32
  print('本次加载模型的设备为GPU: ', torch.cuda.get_device_name(0))
33
  else:
34
  print('本次加载模型的设备为CPU.')
35
- model = SentenceTransformer('./moka-ai_m3e-large',device=device)
36
 
37
  # 创建一个HTTPBearer实例
38
  security = HTTPBearer()
 
32
  print('本次加载模型的设备为GPU: ', torch.cuda.get_device_name(0))
33
  else:
34
  print('本次加载模型的设备为CPU.')
35
+ model = SentenceTransformer('Qwen/Qwen3-Embedding-4B',device=device)
36
 
37
  # 创建一个HTTPBearer实例
38
  security = HTTPBearer()