oyly commited on
Commit
c8919af
·
1 Parent(s): 0123c32

fix: inverse and steps

Browse files
Files changed (1) hide show
  1. app.py +7 -9
app.py CHANGED
@@ -34,7 +34,7 @@ class FluxEditor_lore_demo:
34
 
35
  self.name = model_name
36
  self.is_schnell = model_name == "flux-schnell"
37
- self.resize_longside = 800
38
  self.save = False
39
 
40
  self.output_dir = 'outputs_gradio'
@@ -136,7 +136,7 @@ class FluxEditor_lore_demo:
136
  print(f"inversion Done in {t1 - t0:.1f}s.")
137
  return init_image
138
 
139
- @spaces.GPU(duration=120)
140
  def edit(self, brush_canvas, source_prompt, inversion_guidance,
141
  target_prompt, target_object,target_object_index,
142
  inversion_num_steps, injection_num_steps,
@@ -145,11 +145,9 @@ class FluxEditor_lore_demo:
145
  ):
146
 
147
  torch.cuda.empty_cache()
148
- if 'src_p' not in self.info or self.info['src_p'] != source_prompt:
149
- print('src prompt changed. inverse again')
150
- self.inverse(brush_canvas,source_prompt,
151
- inversion_num_steps, injection_num_steps,
152
- inversion_guidance)
153
 
154
  rgba_init_image = brush_canvas["background"]
155
  rgba_mask = brush_canvas["layers"][0]
@@ -291,13 +289,13 @@ def create_demo(model_name: str):
291
  <b>Here are editing steps:</b> <br>
292
  1️⃣ Upload your source image. <br>
293
  2️⃣ Fill in your source prompt and use the brush tool to draw your mask. (on layer 1) <br>
294
- 3️⃣ Fill in your target prompt, target object and its index in target prompt (index start from 0).
295
  4️⃣ Adjust the hyperparameters. <br>
296
  5️⃣ Click the "Edit" button to generate your edited image! <br>
297
 
298
  🎨 [<b>Examples</b>] Click our examples below, draw your mask and click the "Edit" button. <br>
299
 
300
- 🔔 [<b>Note</b>] Due to limited resources, we will resize image to <=800 longside. <br>
301
  go to https://github.com/oyly16/LORE for more usage. <br>
302
  """
303
  article = r"""
 
34
 
35
  self.name = model_name
36
  self.is_schnell = model_name == "flux-schnell"
37
+ self.resize_longside = 512
38
  self.save = False
39
 
40
  self.output_dir = 'outputs_gradio'
 
136
  print(f"inversion Done in {t1 - t0:.1f}s.")
137
  return init_image
138
 
139
+ @spaces.GPU(duration=60)
140
  def edit(self, brush_canvas, source_prompt, inversion_guidance,
141
  target_prompt, target_object,target_object_index,
142
  inversion_num_steps, injection_num_steps,
 
145
  ):
146
 
147
  torch.cuda.empty_cache()
148
+ self.inverse(brush_canvas,source_prompt,
149
+ inversion_num_steps, injection_num_steps,
150
+ inversion_guidance)
 
 
151
 
152
  rgba_init_image = brush_canvas["background"]
153
  rgba_mask = brush_canvas["layers"][0]
 
289
  <b>Here are editing steps:</b> <br>
290
  1️⃣ Upload your source image. <br>
291
  2️⃣ Fill in your source prompt and use the brush tool to draw your mask. (on layer 1) <br>
292
+ 3️⃣ Fill in your target prompt, target object and its index in target prompt (index start from 0). <br>
293
  4️⃣ Adjust the hyperparameters. <br>
294
  5️⃣ Click the "Edit" button to generate your edited image! <br>
295
 
296
  🎨 [<b>Examples</b>] Click our examples below, draw your mask and click the "Edit" button. <br>
297
 
298
+ 🔔 [<b>Note</b>] Due to limited resources, we will resize image to <=512 longside. <br>
299
  go to https://github.com/oyly16/LORE for more usage. <br>
300
  """
301
  article = r"""