KPrashanth commited on
Commit
8f81473
·
verified ·
1 Parent(s): 0ff83d6

Update core_logic/llm_config.py

Browse files
Files changed (1) hide show
  1. 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 = "TinyLlama/TinyLlama-1.1B-Chat-v1.0" # <-- change if you're using a different repo
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)