abenkbp commited on
Commit
6f492b3
·
1 Parent(s): 639143e
Files changed (1) hide show
  1. data/models/llama3-1-70b-nvidia.py +1 -1
data/models/llama3-1-70b-nvidia.py CHANGED
@@ -16,7 +16,7 @@ client = OpenAI(
16
  # Create the completion
17
  completion = client.chat.completions.create(
18
  model="meta/llama-3.1-70b-instruct",
19
- messages=[{"role":"user","content": args.message}],
20
  temperature=0.2,
21
  top_p=0.7,
22
  max_tokens=1024,
 
16
  # Create the completion
17
  completion = client.chat.completions.create(
18
  model="meta/llama-3.1-70b-instruct",
19
+ messages=[type(args.message)],
20
  temperature=0.2,
21
  top_p=0.7,
22
  max_tokens=1024,