LarFii commited on
Commit
4c55f30
·
1 Parent(s): d76f618

update llm.py

Browse files
Files changed (1) hide show
  1. 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
- input_string = kwargs['hashing_kv'].global_config['llm_model_name']
122
  return await hf_model_if_cache(
123
- input_string,
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,