barreloflube commited on
Commit
39e615b
·
1 Parent(s): 9aa9efa

Refactor image_tab function to update generator initialization and remove unnecessary code

Browse files
Files changed (1) hide show
  1. tabs/images/handlers.py +1 -1
tabs/images/handlers.py CHANGED
@@ -63,7 +63,6 @@ def get_pipe(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq):
63
  pipe_args['pipeline'] = AutoPipelineForText2Image.from_pipe(**pipe_args)
64
  elif request.custom_addons:
65
  pipe_args['pipeline'] = None
66
- gr.Info(f"Pipeline Mode: {str(type(pipe_args['pipeline']))}")
67
 
68
  # Enable or Disable Vae
69
  if request.vae:
@@ -252,6 +251,7 @@ def gen_img(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq, progress=gr.Prog
252
 
253
  # Generate
254
  progress(0.9, "Generating Images")
 
255
  images = pipeline(**args).images
256
 
257
  # Refiner
 
63
  pipe_args['pipeline'] = AutoPipelineForText2Image.from_pipe(**pipe_args)
64
  elif request.custom_addons:
65
  pipe_args['pipeline'] = None
 
66
 
67
  # Enable or Disable Vae
68
  if request.vae:
 
251
 
252
  # Generate
253
  progress(0.9, "Generating Images")
254
+ gr.Info(f"Pipeline args: {args}", duration=60)
255
  images = pipeline(**args).images
256
 
257
  # Refiner