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