leavoigt commited on
Commit
9296721
·
verified ·
1 Parent(s): f17c0cf

Update utils/generator.py

Browse files
Files changed (1) hide show
  1. utils/generator.py +2 -0
utils/generator.py CHANGED
@@ -59,6 +59,8 @@ def get_chat_model():
59
  repo_id=MODEL,
60
  huggingfacehub_api_token=auth_config["api_key"],
61
  task="text-generation",
 
 
62
  temperature=TEMPERATURE,
63
  max_new_tokens=MAX_TOKENS
64
  )
 
59
  repo_id=MODEL,
60
  huggingfacehub_api_token=auth_config["api_key"],
61
  task="text-generation",
62
+ provider='novita',
63
+ server_kwargs={"bill_to": 'GIZ'},
64
  temperature=TEMPERATURE,
65
  max_new_tokens=MAX_TOKENS
66
  )