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