From bdf9db27a7e3b231354b50248321fe25873703f2 Mon Sep 17 00:00:00 2001 From: kqlio67 Date: Tue, 22 Oct 2024 12:35:30 +0300 Subject: Updated provider (g4f/Provider/nexra/NexraDallE.py) --- g4f/Provider/nexra/NexraDallE.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'g4f') diff --git a/g4f/Provider/nexra/NexraDallE.py b/g4f/Provider/nexra/NexraDallE.py index 7b3ac388..dad1d057 100644 --- a/g4f/Provider/nexra/NexraDallE.py +++ b/g4f/Provider/nexra/NexraDallE.py @@ -24,6 +24,7 @@ class NexraDallE(AbstractProvider, ProviderModelMixin): cls, model: str, messages: Messages, + response: str = "url", # base64 or url **kwargs ) -> CreateResult: model = cls.get_model(model) @@ -35,7 +36,7 @@ class NexraDallE(AbstractProvider, ProviderModelMixin): data = { "prompt": messages[-1]["content"], "model": model, - "response": "url" + "response": response } response = requests.post(cls.api_endpoint, headers=headers, json=data) -- cgit v1.2.3