Spaces:
Runtime error
Runtime error
Zwea Htet
commited on
Commit
•
5e8fa58
1
Parent(s):
069e494
update llama custom
Browse files- 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(
|
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 |
)
|