abenkbp commited on
Commit
8eeaa1e
·
1 Parent(s): 68bb8db
Files changed (1) hide show
  1. data/models/llama3-1-8b.py +1 -1
data/models/llama3-1-8b.py CHANGED
@@ -28,7 +28,7 @@ def chat_completion():
28
  try:
29
  print(data.get('messages', []))
30
  outputs = pipeline(
31
- user_input=data.get('messages', []),
32
  max_new_tokens=data.get('max_tokens', 2048),
33
  temperature=data.get('temperature', 0.7),
34
  top_p=data.get('top_p', 0.95)
 
28
  try:
29
  print(data.get('messages', []))
30
  outputs = pipeline(
31
+ text_inputs=data.get('messages', []),
32
  max_new_tokens=data.get('max_tokens', 2048),
33
  temperature=data.get('temperature', 0.7),
34
  top_p=data.get('top_p', 0.95)