ldooks/dragon-bot
ldooks/dragon-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

give me more image generation steps

Browse Source
This commit is contained in:
Luke Robles 2022-10-06 07:57:25 -07:00
parent bf67c31df4
commit 754a7e020f
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/cogs/actual_utils.py
View File

@ -73,10 +73,12 @@ class ActualUtils(commands.Cog):
port = "7860" port = "7860"
ip = "192.168.1.80" ip = "192.168.1.80"
steps = 20
# Send my requests to my gaming computer with the 3080 (if its up) # Send my requests to my gaming computer with the 3080 (if its up)
if ctx.message.author.id == 144986109804412928: if ctx.message.author.id == 144986109804412928:
ip = "192.168.1.188" ip = "192.168.1.188"
steps = 60
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.settimeout(1) s.settimeout(1)
try: try:
@ -93,7 +95,7 @@ class ActualUtils(commands.Cog):
positives = prompt.split(";")[0] positives = prompt.split(";")[0]
negatives = prompt.split(";")[1] negatives = prompt.split(";")[1]
file_path = await stable_diffusion.generate_image( file_path = await stable_diffusion.generate_image(
ip=ip, port=port, positives=positives, negatives=negatives ip=ip, port=port, positives=positives, negatives=negatives, steps=steps
) )
await original_message.edit( await original_message.edit(
content="", content="",

4
app/stable_diffusion.py
View File

@ -5,7 +5,7 @@ import socket
import tempfile import tempfile
async def generate_image(ip, port, positives, negatives): async def generate_image(ip, port, positives, negatives, steps):
url = "http://" + ip + ":" + port + "/api/predict/" url = "http://" + ip + ":" + port + "/api/predict/"
headers = { headers = {
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:105.0) Gecko/20100101 Firefox/105.0", "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.15; rv:105.0) Gecko/20100101 Firefox/105.0",
@ -22,7 +22,7 @@ async def generate_image(ip, port, positives, negatives):
negatives, # Negative Prompt negatives, # Negative Prompt
"None", "None",
"None", "None",
20, # Sampling Steps steps, # Sampling Steps
"Euler a", "Euler a",
True, # enable face fix True, # enable face fix
False, False,