summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/DeepInfra.py
diff options
context:
space:
mode:
authornullstreak <139914347+nullstreak@users.noreply.github.com>2024-02-24 19:36:42 +0100
committernullstreak <139914347+nullstreak@users.noreply.github.com>2024-02-24 19:36:42 +0100
commit0d4f637760dc778e52a2c495a676ae0dcf75ebe6 (patch)
treed2dffb31047dc2e12ee70e51f6ebe8f2afa69131 /g4f/Provider/DeepInfra.py
parentMerge pull request #1622 from hlohaus/gem (diff)
downloadgpt4free-0d4f637760dc778e52a2c495a676ae0dcf75ebe6.tar
gpt4free-0d4f637760dc778e52a2c495a676ae0dcf75ebe6.tar.gz
gpt4free-0d4f637760dc778e52a2c495a676ae0dcf75ebe6.tar.bz2
gpt4free-0d4f637760dc778e52a2c495a676ae0dcf75ebe6.tar.lz
gpt4free-0d4f637760dc778e52a2c495a676ae0dcf75ebe6.tar.xz
gpt4free-0d4f637760dc778e52a2c495a676ae0dcf75ebe6.tar.zst
gpt4free-0d4f637760dc778e52a2c495a676ae0dcf75ebe6.zip
Diffstat (limited to '')
-rw-r--r--g4f/Provider/DeepInfra.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/g4f/Provider/DeepInfra.py b/g4f/Provider/DeepInfra.py
index 183f00ea..f44679ff 100644
--- a/g4f/Provider/DeepInfra.py
+++ b/g4f/Provider/DeepInfra.py
@@ -61,6 +61,8 @@ class DeepInfra(AsyncGeneratorProvider, ProviderModelMixin):
'model' : cls.get_model(model),
'messages': messages,
'temperature': kwargs.get("temperature", 0.7),
+ 'max_tokens': kwargs.get("max_tokens", 512),
+ 'stop': kwargs.get("stop", []),
'stream' : True
}
async with session.post('https://api.deepinfra.com/v1/openai/chat/completions',