Update localembedding.py
Browse files- 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('
|
| 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()
|