Update config.py

config.py

@@ -4,14 +4,10 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from sentence_transformers import SentenceTransformer
 from config import EMBEDDING_MODEL_NAME
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
-# Clase para los modelos (opcional, si deseas utilizar pydantic)
-class Models(BaseModel):
-    embedding_model: SentenceTransformer
-    tokenizer: AutoTokenizer
-    yi_coder_model: AutoModelForCausalLM
-    device: torch.device
+class MyModel(BaseModel):
+    model_config = ConfigDict(arbitrary_types_allowed=True)
 
 # Cargar el modelo de embeddings
 def load_embedding_model():