From a9ba2b2db216ed8e5a13f39d46eae0507a202007 Mon Sep 17 00:00:00 2001 From: Yandrik Date: Thu, 2 May 2024 08:04:05 +0200 Subject: [PATCH] feat(textgen): switched to zukijourney api for textgen --- gen_service/main.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/gen_service/main.py b/gen_service/main.py index 56d35ab..d42ae9b 100644 --- a/gen_service/main.py +++ b/gen_service/main.py @@ -24,7 +24,7 @@ def gen_text(): print(prompt) try: - text = _gen_text(prompt) + text = _gen_text(prompt, use_zukijourney=True) except Exception as e: print(f"Error generating text: {e}") return f"Error generating text: {str(e)}", 500 @@ -61,7 +61,9 @@ anime art, surreal (biomorphic landscape:1.1) on a cloud, art made by complex AI Here's the request: {prompt} {f"\nKeep to the following style: {style}" if style else ""} -Output the prompt in a txt code block.""") +Output the prompt in a txt code block.""", + use_zukijourney=True +) print(gen_prompt) @@ -93,8 +95,8 @@ def _gen_text(prompt, use_zukijourney=False): else: client = get_g4f_client() response = client.chat.completions.create( - provider=g4f.Provider.Llama, - model="llama-3-70b", + provider=g4f.Provider.PerplexityLabs, + model="llama-3-70b-instruct", messages=[{"role": "user", "content": prompt}], max_tokens=800, temperature=0.7,