C2MV commited on
Commit
ec557e5
verified
1 Parent(s): f780443

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +2 -2
models.py CHANGED
@@ -1,4 +1,4 @@
1
- # models.py (Este c贸digo ahora es CORRECTO con el nuevo requirements.txt)
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 ahora funcionar谩 gracias a la actualizaci贸n en requirements.txt
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,