Home/weeabot
2
1
Fork 0
Code Issues 1 Pull Requests Projects Releases Wiki Activity

Fix typos

Browse Source
This commit is contained in:
jordan 2022-12-10 22:11:44 +11:00
parent 991c592086
commit ac3485d59e
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
DiffuseAPI.py
View File

@ -43,7 +43,7 @@ class DiffuseAPI():
return False return False
def set_styles(self, styles): def set_styles(self, styles):
if type styles = list: if type(styles) == list:
self.styles = styles self.styles = styles
return True return True
return False return False
@ -74,7 +74,7 @@ class DiffuseAPI():
return ("portrait", self.width, self.height) return ("portrait", self.width, self.height)
def set_nsfw_filter(self, filter_state): def set_nsfw_filter(self, filter_state):
if type filter_state = bool: if type(filter_state) == bool:
self.nsfw_enabled = filter_state self.nsfw_enabled = filter_state
return True return True
return False return False
@ -85,9 +85,9 @@ class DiffuseAPI():
async def generate_image(self, prompt, neg_prompt=""): async def generate_image(self, prompt, neg_prompt=""):
payload = { payload = {
"prompt": prompt, "prompt": prompt,
"styles": main_settings.styles, "styles": self.styles,
"steps": main_settings.num_steps, "steps": self.num_steps,
"seed": main_settings.ai_seed, "seed": self.ai_seed,
"n_iter": 1, "n_iter": 1,
"height": self.height, "height": self.height,
"width": self.width, "width": self.width,
@ -105,7 +105,7 @@ class DiffuseAPI():
payload.update(override_payload) payload.update(override_payload)
sess = aiohttp.ClientSession(main_settings.url) sess = aiohttp.ClientSession(self.url)
alive = await sess.head('/') alive = await sess.head('/')
if alive.status != 200: if alive.status != 200:

8
classweeb.ipynb
View File

@ -100,9 +100,9 @@
" async def generate_image(self, prompt, neg_prompt=\"\"):\n", " async def generate_image(self, prompt, neg_prompt=\"\"):\n",
" payload = {\n", " payload = {\n",
" \"prompt\": prompt,\n", " \"prompt\": prompt,\n",
" \"styles\": main_settings.styles,\n", " \"styles\": self.styles,\n",
" \"steps\": main_settings.num_steps,\n", " \"steps\": self.num_steps,\n",
" \"seed\": main_settings.ai_seed,\n", " \"seed\": self.ai_seed,\n",
" \"n_iter\": 1,\n", " \"n_iter\": 1,\n",
" \"height\": self.height,\n", " \"height\": self.height,\n",
" \"width\": self.width,\n", " \"width\": self.width,\n",
@ -120,7 +120,7 @@
"\n", "\n",
" payload.update(override_payload)\n", " payload.update(override_payload)\n",
"\n", "\n",
" sess = aiohttp.ClientSession(main_settings.url)\n", " sess = aiohttp.ClientSession(self.url)\n",
" alive = await sess.head('/')\n", " alive = await sess.head('/')\n",
"\n", "\n",
" if alive.status != 200:\n", " if alive.status != 200:\n",