barakmeiri commited on
Commit
3539851
·
verified ·
1 Parent(s): 3a6432f

convert str to number for the examples log.csv

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -60,9 +60,18 @@ with gr.Blocks(css="style.css") as demo:
60
  @spaces.GPU
61
  def set_pipe(image_editor, input_image, description_prompt, edit_guidance_scale, num_inference_steps=4,
62
  num_inversion_steps=4, inversion_max_step=0.6, rnri_iterations=2, rnri_alpha=0.1, rnri_lr=0.2):
63
- if image_editor is not None:
64
- image_editor = image_editor.to('cpu')
65
-
 
 
 
 
 
 
 
 
 
66
  config = RunConfig(num_inference_steps=num_inference_steps,
67
  num_inversion_steps=num_inversion_steps,
68
  edit_guidance_scale=edit_guidance_scale,
 
60
  @spaces.GPU
61
  def set_pipe(image_editor, input_image, description_prompt, edit_guidance_scale, num_inference_steps=4,
62
  num_inversion_steps=4, inversion_max_step=0.6, rnri_iterations=2, rnri_alpha=0.1, rnri_lr=0.2):
63
+ # if image_editor is not None:
64
+ # image_editor = image_editor.to('cpu')
65
+
66
+ num_inference_steps = int(num_inference_steps) if isinstance(num_inference_steps, str)
67
+ num_inversion_steps = int(num_inversion_steps) if isinstance(num_inversion_steps, str)
68
+ edit_guidance_scale = float(edit_guidance_scale) if isinstance(edit_guidance_scale, str)
69
+ inversion_max_step = float(inversion_max_step) if isinstance(inversion_max_step, str)
70
+ rnri_iterations = int(rnri_iterations) if isinstance(rnri_iterations, str)
71
+ rnri_alpha = float(rnri_alpha) if isinstance(rnri_alpha, str)
72
+ rnri_lr = float(rnri_lr) if isinstance(rnri_lr, str)
73
+
74
+
75
  config = RunConfig(num_inference_steps=num_inference_steps,
76
  num_inversion_steps=num_inversion_steps,
77
  edit_guidance_scale=edit_guidance_scale,