acumplido commited on
Commit
589dd3e
·
verified ·
1 Parent(s): 7164484

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -113,7 +113,7 @@ class RAG:
113
  chat_completion = client.chat.completions.create(
114
  model="tgi",
115
  messages=[
116
- {"role": "user", "content": instruction}
117
  ],
118
  temperature=model_parameters["temperature"],
119
  max_tokens=model_parameters["max_new_tokens"],
 
113
  chat_completion = client.chat.completions.create(
114
  model="tgi",
115
  messages=[
116
+ {"role": "user", "content": query}
117
  ],
118
  temperature=model_parameters["temperature"],
119
  max_tokens=model_parameters["max_new_tokens"],