GuglielmoTor commited on
Commit
7c75922
·
verified ·
1 Parent(s): ec6c545

Update eb_agent_module.py

Browse files
Files changed (1) hide show
  1. eb_agent_module.py +1 -1
eb_agent_module.py CHANGED
@@ -330,7 +330,7 @@ class PandasLLM:
330
  self.generative_model_service.generate_content,
331
  model=model_id_for_api,
332
  contents=contents_for_api,
333
- generation_config=generation_config_to_pass,
334
  safety_settings=safety_settings_to_pass
335
  )
336
 
 
330
  self.generative_model_service.generate_content,
331
  model=model_id_for_api,
332
  contents=contents_for_api,
333
+ config=generation_config_to_pass,
334
  safety_settings=safety_settings_to_pass
335
  )
336