phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Removed combine_dict function as it isn't needed

Browse source
This commit is contained in:
phixxy 2023-07-25 17:19:13 -07:00
parent de1b78149c
commit 1f3c5cf04a
1 changed files with 2 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

12
sparkytron3000.py
View file

@ -162,14 +162,6 @@ def extract_key_value_pairs(input_str):
return key_value_pairs, output_str return key_value_pairs, output_str
def combine_dicts(dict1, dict2): #prioritizes dict2 args
combined_dict = {}
for key in dict1:
combined_dict[key] = dict1[key]
for key in dict2:
combined_dict[key] = dict2[key]
return combined_dict
def my_open_img_file(path): def my_open_img_file(path):
img = Image.open(path) img = Image.open(path)
w, h = img.size w, h = img.size
@ -1361,7 +1353,7 @@ async def imagine(ctx):
headers = { headers = {
'Content-Type': 'application/json' 'Content-Type': 'application/json'
} }
payload = combine_dicts(payload, key_value_pairs) payload.update(key_value_pairs)
try: try:
async with bot.http_session.post(url, headers=headers, json=payload) as resp: async with bot.http_session.post(url, headers=headers, json=payload) as resp:
@ -1496,7 +1488,7 @@ async def reimagine(ctx):
await ctx.send("Please be patient this may take some time! Generating: " + prompt + ".") await ctx.send("Please be patient this may take some time! Generating: " + prompt + ".")
payload = {"init_images": [img_link], "prompt": prompt, "steps": 40, "negative_prompt": negative_prompt, "denoising_strength": 0.5} payload = {"init_images": [img_link], "prompt": prompt, "steps": 40, "negative_prompt": negative_prompt, "denoising_strength": 0.5}
payload = combine_dicts(payload, key_value_pairs) payload.update(key_value_pairs)
try: try:
async with bot.http_session.post(url=f'{url}/sdapi/v1/img2img', json=payload) as response: async with bot.http_session.post(url=f'{url}/sdapi/v1/img2img', json=payload) as response: