(
self,
# Prompt
prompt,
negative_prompt="",
cfg_scale=1.0,
embedded_guidance=3.5,
t5_sequence_length=512,
# Image
input_image=None,
denoising_strength=1.0,
height=1024,
width=1024,
seed=None,
# Steps
num_inference_steps=30,
# local prompts
local_prompts=(),
masks=(),
mask_scales=(),
# ControlNet
controlnet_image=None,
controlnet_inpaint_mask=None,
enable_controlnet_on_negative=False,
# IP-Adapter
ipadapter_images=None,
ipadapter_scale=1.0,
# EliGen
eligen_entity_prompts=None,
eligen_entity_masks=None,
enable_eligen_on_negative=False,
enable_eligen_inpaint=False,
# InfiniteYou
infinityou_id_image=None,
infinityou_guidance=1.0,
# Flex
flex_inpaint_image=None,
flex_inpaint_mask=None,
flex_control_image=None,
flex_control_strength=0.5,
flex_control_stop=0.5,
# Step1x
step1x_reference_image=None,
# TeaCache
tea_cache_l1_thresh=None,
# Tile
tiled=False,
tile_size=128,
tile_stride=64,
# Progress bar
progress_bar_cmd=tqdm,
progress_bar_st=None,
)
| 419 | |
| 420 | @torch.no_grad() |
| 421 | def __call__( |
| 422 | self, |
| 423 | # Prompt |
| 424 | prompt, |
| 425 | negative_prompt="", |
| 426 | cfg_scale=1.0, |
| 427 | embedded_guidance=3.5, |
| 428 | t5_sequence_length=512, |
| 429 | # Image |
| 430 | input_image=None, |
| 431 | denoising_strength=1.0, |
| 432 | height=1024, |
| 433 | width=1024, |
| 434 | seed=None, |
| 435 | # Steps |
| 436 | num_inference_steps=30, |
| 437 | # local prompts |
| 438 | local_prompts=(), |
| 439 | masks=(), |
| 440 | mask_scales=(), |
| 441 | # ControlNet |
| 442 | controlnet_image=None, |
| 443 | controlnet_inpaint_mask=None, |
| 444 | enable_controlnet_on_negative=False, |
| 445 | # IP-Adapter |
| 446 | ipadapter_images=None, |
| 447 | ipadapter_scale=1.0, |
| 448 | # EliGen |
| 449 | eligen_entity_prompts=None, |
| 450 | eligen_entity_masks=None, |
| 451 | enable_eligen_on_negative=False, |
| 452 | enable_eligen_inpaint=False, |
| 453 | # InfiniteYou |
| 454 | infinityou_id_image=None, |
| 455 | infinityou_guidance=1.0, |
| 456 | # Flex |
| 457 | flex_inpaint_image=None, |
| 458 | flex_inpaint_mask=None, |
| 459 | flex_control_image=None, |
| 460 | flex_control_strength=0.5, |
| 461 | flex_control_stop=0.5, |
| 462 | # Step1x |
| 463 | step1x_reference_image=None, |
| 464 | # TeaCache |
| 465 | tea_cache_l1_thresh=None, |
| 466 | # Tile |
| 467 | tiled=False, |
| 468 | tile_size=128, |
| 469 | tile_stride=64, |
| 470 | # Progress bar |
| 471 | progress_bar_cmd=tqdm, |
| 472 | progress_bar_st=None, |
| 473 | ): |
| 474 | height, width = self.check_resize_height_width(height, width) |
| 475 | |
| 476 | # Tiler parameters |
| 477 | tiler_kwargs = {"tiled": tiled, "tile_size": tile_size, "tile_stride": tile_stride} |
| 478 |
nothing calls this directly
no test coverage detected