LarFii
commited on
Commit
·
4c55f30
1
Parent(s):
d76f618
update llm.py
Browse files- lightrag/llm.py +2 -2
lightrag/llm.py
CHANGED
|
@@ -118,9 +118,9 @@ async def gpt_4o_mini_complete(
|
|
| 118 |
async def hf_model_complete(
|
| 119 |
prompt, system_prompt=None, history_messages=[], **kwargs
|
| 120 |
) -> str:
|
| 121 |
-
|
| 122 |
return await hf_model_if_cache(
|
| 123 |
-
|
| 124 |
prompt,
|
| 125 |
system_prompt=system_prompt,
|
| 126 |
history_messages=history_messages,
|
|
|
|
| 118 |
async def hf_model_complete(
|
| 119 |
prompt, system_prompt=None, history_messages=[], **kwargs
|
| 120 |
) -> str:
|
| 121 |
+
model_name = kwargs['hashing_kv'].global_config['llm_model_name']
|
| 122 |
return await hf_model_if_cache(
|
| 123 |
+
model_name,
|
| 124 |
prompt,
|
| 125 |
system_prompt=system_prompt,
|
| 126 |
history_messages=history_messages,
|