pgarbacki commited on
Commit
487cbc0
1 Parent(s): e59cc56

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -101,14 +101,13 @@ function_spec = [
101
  functions = json.dumps(function_spec, indent=4)
102
 
103
  messages = [
104
- {'role': 'functions', 'content': functions},
105
  {'role': 'system', 'content': 'You are a helpful assistant with access to functions. Use them if required.'},
106
  {'role': 'user', 'content': 'Hi, can you tell me the current stock price of google and netflix?'}
107
  ]
108
 
109
  now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
110
 
111
- model_inputs = tokenizer.apply_chat_template(messages, datetime=now, return_tensors="pt").to(model.device)
112
 
113
  generated_ids = model.generate(model_inputs, max_new_tokens=128)
114
  decoded = tokenizer.batch_decode(generated_ids)
 
101
  functions = json.dumps(function_spec, indent=4)
102
 
103
  messages = [
 
104
  {'role': 'system', 'content': 'You are a helpful assistant with access to functions. Use them if required.'},
105
  {'role': 'user', 'content': 'Hi, can you tell me the current stock price of google and netflix?'}
106
  ]
107
 
108
  now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
109
 
110
+ model_inputs = tokenizer.apply_chat_template(messages, functions=functions, datetime=now, return_tensors="pt").to(model.device)
111
 
112
  generated_ids = model.generate(model_inputs, max_new_tokens=128)
113
  decoded = tokenizer.batch_decode(generated_ids)