bmarci commited on
Commit
af56a7b
·
1 Parent(s): 1ce1a03
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -45,17 +45,18 @@ def infer(
45
  return None
46
  IMG_SIZE = 512
47
  image = pipeline.generate_image(
48
- captions=editing_caption,
 
49
  num_images_per_caption=1,
50
  positive_prompt=positive_prompt,
51
  negative_prompt=negative_prompt,
52
- hw=(IMG_SIZE, IMG_SIZE),
53
  cfg=text_cfg,
54
  cfg_img=img_cfg,
55
  cfg_schedule="constant",
56
- use_norm=True,
57
  num_sampling_steps=num_inference_steps,
58
- seed=seed,
 
59
  progress=True,
60
  )
61
 
 
45
  return None
46
  IMG_SIZE = 512
47
  image = pipeline.generate_image(
48
+ editing_caption,
49
+ hw=(IMG_SIZE, IMG_SIZE),
50
  num_images_per_caption=1,
51
  positive_prompt=positive_prompt,
52
  negative_prompt=negative_prompt,
 
53
  cfg=text_cfg,
54
  cfg_img=img_cfg,
55
  cfg_schedule="constant",
56
+ use_norm=False,
57
  num_sampling_steps=num_inference_steps,
58
+ timesteps_shift=1.0,
59
+ seed=3407,
60
  progress=True,
61
  )
62