Zwea Htet commited on
Commit
5e8fa58
1 Parent(s): 069e494

update llama custom

Browse files
Files changed (1) hide show
  1. models/llamaCustom.py +1 -1
models/llamaCustom.py CHANGED
@@ -112,7 +112,7 @@ class LlamaCustom:
112
  chunk_overlap_ratio=CHUNK_OVERLAP_RATION,
113
  )
114
 
115
- llm_predictor = LLMPredictor(llm=CustomLLM(model_name=model_name))
116
  service_context = ServiceContext.from_defaults(
117
  llm_predictor=llm_predictor, prompt_helper=prompt_helper
118
  )
 
112
  chunk_overlap_ratio=CHUNK_OVERLAP_RATION,
113
  )
114
 
115
+ llm_predictor = LLMPredictor(llm=CustomLLM(llm_model_name=model_name))
116
  service_context = ServiceContext.from_defaults(
117
  llm_predictor=llm_predictor, prompt_helper=prompt_helper
118
  )