Spaces:
Sleeping
Sleeping
Prueba 7b
Browse files- src/model_load.py +1 -1
src/model_load.py
CHANGED
|
@@ -8,7 +8,7 @@ from langchain.chains import LLMChain
|
|
| 8 |
|
| 9 |
def load_model():
|
| 10 |
|
| 11 |
-
model_name="tiiuae/Falcon3-
|
| 12 |
max_memory = {0: "24GB", "cpu": "30GB"}
|
| 13 |
# Cargar tokenizer y modelo de Hugging Face
|
| 14 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
|
| 8 |
|
| 9 |
def load_model():
|
| 10 |
|
| 11 |
+
model_name="tiiuae/Falcon3-7B-Instruct"
|
| 12 |
max_memory = {0: "24GB", "cpu": "30GB"}
|
| 13 |
# Cargar tokenizer y modelo de Hugging Face
|
| 14 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|