Spaces:
Sleeping
Sleeping
Update models.py
Browse files
models.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
# models.py
|
2 |
|
3 |
import torch
|
4 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
@@ -10,7 +10,7 @@ def load_embedding_model():
|
|
10 |
device_str = 'cuda' if torch.cuda.is_available() else 'cpu'
|
11 |
device = torch.device(device_str)
|
12 |
|
13 |
-
# Esta l铆nea
|
14 |
embedding_model = SentenceTransformer(
|
15 |
EMBEDDING_MODEL_NAME,
|
16 |
device=device,
|
|
|
1 |
+
# models.py
|
2 |
|
3 |
import torch
|
4 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
|
10 |
device_str = 'cuda' if torch.cuda.is_available() else 'cpu'
|
11 |
device = torch.device(device_str)
|
12 |
|
13 |
+
# Esta l铆nea ES CORRECTA y debe coincidir con el requirements.txt actualizado.
|
14 |
embedding_model = SentenceTransformer(
|
15 |
EMBEDDING_MODEL_NAME,
|
16 |
device=device,
|