Yandrik/lottis_imagen
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: switch provider and add zukijourney textgen parameter

Browse Source
This commit is contained in:
Yandrik 2024-05-02 07:57:23 +02:00
parent b1445e681c
commit 30fb94c2de
1 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
gen_service/main.py
View File

@ -80,10 +80,20 @@ Output the prompt in a txt code block.""")
def _gen_text(prompt):
def _gen_text(prompt, use_zukijourney=False):
if use_zukijourney:
client = get_client()
response = client.chat.completions.create(
model="llama-3-70b-instruct",
messages=[{"role": "user", "content": prompt}],
max_tokens=800,
temperature=0.7,
)
else:
client = get_g4f_client() client = get_g4f_client()
response = client.chat.completions.create( response = client.chat.completions.create(
provider=g4f.Provider.PerplexityLabs, provider=g4f.Provider.Llama,
model="llama-3-70b-instruct", model="llama-3-70b-instruct",
messages=[{"role": "user", "content": prompt}], messages=[{"role": "user", "content": prompt}],
max_tokens=800, max_tokens=800,