Akjava commited on
Commit
3b5fe99
1 Parent(s): 3ca75d4
Files changed (1) hide show
  1. flux1_img2img.py +2 -0
flux1_img2img.py CHANGED
@@ -9,7 +9,9 @@ import spaces
9
 
10
  @spaces.GPU
11
  def process_image(image,mask_image,prompt="a person",model_id="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,num_inference_steps=4):
 
12
  if image == None:
 
13
  return None
14
 
15
  pipe = FluxImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)
 
9
 
10
  @spaces.GPU
11
  def process_image(image,mask_image,prompt="a person",model_id="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,num_inference_steps=4):
12
+ print("start process image process_image")
13
  if image == None:
14
+ print("empty input image returned")
15
  return None
16
 
17
  pipe = FluxImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.bfloat16)