Commit
·
39e615b
1
Parent(s):
9aa9efa
Refactor image_tab function to update generator initialization and remove unnecessary code
Browse files- 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
|