Functions — github.com/FireRedTeam/LayerDiffuse-Flux

hub / github.com/FireRedTeam/LayerDiffuse-Flux / functions

Functions43 in github.com/FireRedTeam/LayerDiffuse-Flux

⨍ Functions 43 ◇ Types & classes 6

↓ 6 callersMethoddecode

(self, latent, aug=True)

lib_layerdiffuse/vae.py:267

↓ 5 callersMethodencode

(self, img_rgba, img_rgb, padded_img_rgb, use_offset=True)

lib_layerdiffuse/vae.py:280

↓ 4 callersMethod__init__

( self, in_channels: int = 3, out_channels: int = 3, down_block_types: Tuple[s

lib_layerdiffuse/vae.py:58

↓ 3 callersMethod_unpack_latents

(latents, height, width, vae_scale_factor)

lib_layerdiffuse/pipeline_flux_img2img.py:507

↓ 2 callersFunctiondist_sample_deterministic

(dist: DiagonalGaussianDistribution, perturbation: torch.Tensor)

lib_layerdiffuse/vae.py:229

↓ 2 callersFunctionpad_rgb

(np_rgba_hwc_uint8)

lib_layerdiffuse/vae.py:211

↓ 2 callersFunctionretrieve_latents

( encoder_output: torch.Tensor, generator: Optional[torch.Generator] = None, sample_mode: str = "sample" )

lib_layerdiffuse/pipeline_flux_img2img.py:89

↓ 2 callersFunctionzero_module

Zero out the parameters of a module and return it.

lib_layerdiffuse/vae.py:19

↓ 1 callersMethod_encode_vae_image

(self, image: torch.Tensor, generator: torch.Generator)

lib_layerdiffuse/pipeline_flux_img2img.py:401

↓ 1 callersMethod_get_clip_prompt_embeds

( self, prompt: Union[str, List[str]], num_images_per_prompt: int = 1, device:

lib_layerdiffuse/pipeline_flux_img2img.py:276

↓ 1 callersMethod_get_t5_prompt_embeds

( self, prompt: Union[str, List[str]] = None, num_images_per_prompt: int = 1,

lib_layerdiffuse/pipeline_flux_img2img.py:226

↓ 1 callersMethod_pack_latents

(latents, batch_size, num_channels_latents, height, width)

lib_layerdiffuse/pipeline_flux_img2img.py:498

↓ 1 callersMethod_prepare_latent_image_ids

(batch_size, height, width, device, dtype)

lib_layerdiffuse/pipeline_flux_img2img.py:483

↓ 1 callersFunctionbuild_alpha_pyramid

(color, alpha, dk=1.2)

lib_layerdiffuse/vae.py:191

↓ 1 callersFunctioncalculate_shift

( image_seq_len, base_seq_len: int = 256, max_seq_len: int = 4096, base_shift: float = 0.5,

lib_layerdiffuse/pipeline_flux_img2img.py:75

↓ 1 callersMethodcheck_inputs

( self, prompt, prompt_2, strength, height, width, pro

lib_layerdiffuse/pipeline_flux_img2img.py:427

↓ 1 callersFunctioncheckerboard

(shape)

lib_layerdiffuse/vae.py:187

↓ 1 callersMethoddevice

(self)

lib_layerdiffuse/vae.py:298

↓ 1 callersMethodencode_prompt

r""" Args: prompt (`str` or `List[str]`, *optional*): prompt to be encoded prompt_2 (`str` or `List[s

lib_layerdiffuse/pipeline_flux_img2img.py:321

↓ 1 callersMethodestimate_augmented

(self, pixel, latent)

lib_layerdiffuse/vae.py:302

↓ 1 callersMethodestimate_augmented

(self, pixel, latent)

lib_layerdiffuse/vae.py:352

↓ 1 callersMethodestimate_single_pass

(self, pixel, latent)

lib_layerdiffuse/vae.py:347

↓ 1 callersFunctiongenerate_img

(pipe, trans_vae, args)

demo_i2i.py:14

↓ 1 callersFunctiongenerate_img

(pipe, trans_vae, args)

demo_t2i.py:10

↓ 1 callersMethodget_timesteps

(self, num_inference_steps, strength, device)

lib_layerdiffuse/pipeline_flux_img2img.py:416

↓ 1 callersMethodprepare_latents

( self, image, timestep, batch_size, num_channels_latents, hei

lib_layerdiffuse/pipeline_flux_img2img.py:522

↓ 1 callersFunctionretrieve_timesteps

r""" Calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. Handles custom timesteps. Any kwa

lib_layerdiffuse/pipeline_flux_img2img.py:103

Method__call__

r""" Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*):

lib_layerdiffuse/pipeline_flux_img2img.py:590

Method__call__

(self, x)

lib_layerdiffuse/vae.py:51

Method__init__

( self, scheduler: FlowMatchEulerDiscreteScheduler, vae: AutoencoderKL, text_e

lib_layerdiffuse/pipeline_flux_img2img.py:193

Method__init__

(self, latent_c=4, *args, **kwargs)

lib_layerdiffuse/vae.py:29

Method__init__

(self, sd_vae, dtype=torch.float16, encoder_file=None, decoder_file=None, alpha=300.0, latent_c=16, *args, **k

lib_layerdiffuse/vae.py:235

Method__init__

(self, filename, dtype=torch.float16, *args, **kwargs)

lib_layerdiffuse/vae.py:335

Method__init__

(self, filename, dtype=torch.float16, alpha=300.0, *args, **kwargs)

lib_layerdiffuse/vae.py:415

Methodforward

(self, x, latent)

lib_layerdiffuse/vae.py:161

Methodforward

(self, img_rgba, img_rgb, padded_img_rgb, use_offset=True)

lib_layerdiffuse/vae.py:294

Methodforward

(self, sd_vae, latent)

lib_layerdiffuse/vae.py:383

Methodforward

(self, sd_vae, list_of_np_rgba_hwc_uint8, use_offset=True)

lib_layerdiffuse/vae.py:432

Methodguidance_scale

(self)

lib_layerdiffuse/pipeline_flux_img2img.py:573

Methodinterrupt

(self)

lib_layerdiffuse/pipeline_flux_img2img.py:585

Methodjoint_attention_kwargs

(self)

lib_layerdiffuse/pipeline_flux_img2img.py:577

Methodnum_timesteps

(self)

lib_layerdiffuse/pipeline_flux_img2img.py:581

Methodsd_decode

(self, latent)

lib_layerdiffuse/vae.py:264