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 |
)
|