Spaces:
Sleeping
Sleeping
Update utils/generator.py
Browse files- 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 |
)
|