VcRlAgent commited on
Commit
9d063c3
·
1 Parent(s): 5c0d061

Starter LLM Inference Call

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ llm = HuggingFaceEndpoint(
59
  repo_id="meta-llama/Llama-3.1-8B-Instruct",
60
  huggingfacehub_api_token=os.environ["HF_TOKEN"],
61
  temperature=0.1,
62
- max_tokens=512
63
  )
64
 
65
  agent = create_sql_agent(
 
59
  repo_id="meta-llama/Llama-3.1-8B-Instruct",
60
  huggingfacehub_api_token=os.environ["HF_TOKEN"],
61
  temperature=0.1,
62
+ max_new_tokens=512
63
  )
64
 
65
  agent = create_sql_agent(