radames commited on
Commit
8fb30db
1 Parent(s): 411c3fa

add negative prompt syntax parsing

Browse files
Files changed (1) hide show
  1. stablediffusion-infinity/app.py +8 -3
stablediffusion-infinity/app.py CHANGED
@@ -138,7 +138,11 @@ async def run_outpaint(
138
  mask = sel_buffer[:, :, -1]
139
  nmask = 255 - mask
140
  process_size = 512
141
-
 
 
 
 
142
  if nmask.sum() < 1:
143
  print("inpaiting with fixed Mask")
144
  mask = np.array(STATIC_MASK)[:, :, 0]
@@ -171,7 +175,8 @@ async def run_outpaint(
171
  # mask_image=mask_image.filter(ImageFilter.GaussianBlur(radius = 8))
172
  with autocast("cuda"):
173
  output = inpaint(
174
- prompt=prompt_text,
 
175
  image=init_image.resize(
176
  (process_size, process_size), resample=SAMPLING_MODE
177
  ),
@@ -188,7 +193,7 @@ async def run_outpaint(
188
 
189
  if not is_nsfw:
190
  # print("not nsfw, uploading")
191
- image_url = await upload_file(image, prompt_text, room_id, image_key)
192
 
193
  params = {
194
  "is_nsfw": is_nsfw,
 
138
  mask = sel_buffer[:, :, -1]
139
  nmask = 255 - mask
140
  process_size = 512
141
+ negative_syntax = r'\<(.*?)\>'
142
+ prompt = re.sub(negative_syntax, ' ', prompt_text)
143
+ negative_prompt = ' '.join(re.findall(negative_syntax, prompt_text))
144
+ print("prompt", prompt)
145
+ print("negative_prompt", negative_prompt)
146
  if nmask.sum() < 1:
147
  print("inpaiting with fixed Mask")
148
  mask = np.array(STATIC_MASK)[:, :, 0]
 
175
  # mask_image=mask_image.filter(ImageFilter.GaussianBlur(radius = 8))
176
  with autocast("cuda"):
177
  output = inpaint(
178
+ prompt=prompt,
179
+ negative_prompt=negative_prompt,
180
  image=init_image.resize(
181
  (process_size, process_size), resample=SAMPLING_MODE
182
  ),
 
193
 
194
  if not is_nsfw:
195
  # print("not nsfw, uploading")
196
+ image_url = await upload_file(image, prompt + "NNOTN" + negative_prompt, room_id, image_key)
197
 
198
  params = {
199
  "is_nsfw": is_nsfw,