Akjava commited on
Commit
9972208
1 Parent(s): e1e11cc

add guidance

Browse files
Files changed (1) hide show
  1. flux1_img2img.py +2 -1
flux1_img2img.py CHANGED
@@ -19,7 +19,8 @@ def process_image(image,mask_image,prompt="a person",model_id="black-forest-labs
19
  generator = torch.Generator("cuda").manual_seed(seed)
20
  generators.append(generator)
21
  # more parameter see https://huggingface.co/docs/diffusers/api/pipelines/flux#diffusers.FluxInpaintPipeline
22
- output = pipe(prompt=prompt, image=image,generator=generator,strength=strength)
 
23
 
24
  # TODO support mask
25
  return output.images[0]
 
19
  generator = torch.Generator("cuda").manual_seed(seed)
20
  generators.append(generator)
21
  # more parameter see https://huggingface.co/docs/diffusers/api/pipelines/flux#diffusers.FluxInpaintPipeline
22
+ output = pipe(prompt=prompt, image=image,generator=generator,strength=strength
23
+ ,guidance_scale=0,num_inference_steps=num_inference_steps)
24
 
25
  # TODO support mask
26
  return output.images[0]