Spaces:
Sleeping
Sleeping
Update core_logic/llm_config.py
Browse files- core_logic/llm_config.py +1 -1
core_logic/llm_config.py
CHANGED
|
@@ -13,7 +13,7 @@ from langchain_huggingface import ChatHuggingFace, HuggingFacePipeline
|
|
| 13 |
from transformers import BitsAndBytesConfig
|
| 14 |
|
| 15 |
|
| 16 |
-
MODEL_ID = "
|
| 17 |
|
| 18 |
|
| 19 |
@lru_cache(maxsize=1)
|
|
|
|
| 13 |
from transformers import BitsAndBytesConfig
|
| 14 |
|
| 15 |
|
| 16 |
+
MODEL_ID = "google/gemma-2-2b-it" # <-- change if you're using a different repo
|
| 17 |
|
| 18 |
|
| 19 |
@lru_cache(maxsize=1)
|