Spaces:

pmelnechuk
/

MathQA

Sleeping

pmelnechuk commited on Feb 21

Commit

2f94c75

verified ·

1 Parent(s): 881820d

Prueba 7b

Files changed (1) hide show

src/model_load.py CHANGED Viewed

@@ -8,7 +8,7 @@ from langchain.chains import LLMChain
 def load_model():
-    model_name="tiiuae/Falcon3-10B-Instruct"
     max_memory = {0: "24GB", "cpu": "30GB"}
     # Cargar tokenizer y modelo de Hugging Face
     tokenizer = AutoTokenizer.from_pretrained(model_name)

 def load_model():
+    model_name="tiiuae/Falcon3-7B-Instruct"
     max_memory = {0: "24GB", "cpu": "30GB"}
     # Cargar tokenizer y modelo de Hugging Face
     tokenizer = AutoTokenizer.from_pretrained(model_name)