SYMBOL INDEX (173 symbols across 15 files)

FILE: demo_gradio.py
  function worker (line 103) | def worker(input_image, prompt, n_prompt, seed, total_second_length, lat...
  function process (line 318) | def process(input_image, prompt, n_prompt, seed, total_second_length, la...
  function end_process (line 346) | def end_process():

FILE: demo_gradio_f1.py
  function worker (line 103) | def worker(input_image, prompt, n_prompt, seed, total_second_length, lat...
  function process (line 300) | def process(input_image, prompt, n_prompt, seed, total_second_length, la...
  function end_process (line 328) | def end_process():

FILE: diffusers_helper/bucket_tools.py
  function find_nearest_bucket (line 21) | def find_nearest_bucket(h, w, resolution=640):

FILE: diffusers_helper/clip_vision.py
  function hf_clip_vision_encode (line 4) | def hf_clip_vision_encode(image, feature_extractor, image_encoder):

FILE: diffusers_helper/dit_common.py
  function LayerNorm_forward (line 10) | def LayerNorm_forward(self, x):
  function FP32LayerNorm_forward (line 18) | def FP32LayerNorm_forward(self, x):
  function RMSNorm_forward (line 32) | def RMSNorm_forward(self, hidden_states):
  function AdaLayerNormContinuous_forward (line 46) | def AdaLayerNormContinuous_forward(self, x, conditioning_embedding):

FILE: diffusers_helper/gradio/progress_bar.py
  function make_progress_bar_html (line 81) | def make_progress_bar_html(number, text):
  function make_progress_bar_css (line 85) | def make_progress_bar_css():

FILE: diffusers_helper/hf_login.py
  function login (line 4) | def login(token):

FILE: diffusers_helper/hunyuan.py
  function encode_prompt_conds (line 8) | def encode_prompt_conds(prompt, text_encoder, text_encoder_2, tokenizer,...
  function vae_decode_fake (line 62) | def vae_decode_fake(latents):
  function vae_decode (line 94) | def vae_decode(latents, vae, image_mode=False):
  function vae_encode (line 108) | def vae_encode(image, vae):

FILE: diffusers_helper/k_diffusion/uni_pc_fm.py
  function expand_dims (line 12) | def expand_dims(v, dims):
  class FlowMatchUniPC (line 16) | class FlowMatchUniPC:
    method __init__ (line 17) | def __init__(self, model, extra_args, variant='bh1'):
    method model_fn (line 22) | def model_fn(self, x, t):
    method update_fn (line 25) | def update_fn(self, x, model_prev_list, t_prev_list, t, order):
    method sample (line 111) | def sample(self, x, sigmas, callback=None, disable_pbar=False):
  function sample_unipc (line 139) | def sample_unipc(model, noise, sigmas, extra_args=None, callback=None, d...

FILE: diffusers_helper/k_diffusion/wrapper.py
  function append_dims (line 4) | def append_dims(x, target_dims):
  function rescale_noise_cfg (line 8) | def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=1.0):
  function fm_wrapper (line 19) | def fm_wrapper(transformer, t_scale=1000.0):

FILE: diffusers_helper/memory.py
  class DynamicSwapInstaller (line 12) | class DynamicSwapInstaller:
    method _install_module (line 14) | def _install_module(module: torch.nn.Module, **kwargs):
    method _uninstall_module (line 42) | def _uninstall_module(module: torch.nn.Module):
    method install_model (line 48) | def install_model(model: torch.nn.Module, **kwargs):
    method uninstall_model (line 54) | def uninstall_model(model: torch.nn.Module):
  function fake_diffusers_current_device (line 60) | def fake_diffusers_current_device(model: torch.nn.Module, target_device:...
  function get_cuda_free_memory_gb (line 71) | def get_cuda_free_memory_gb(device=None):
  function move_model_to_device_with_memory_preservation (line 84) | def move_model_to_device_with_memory_preservation(model, target_device, ...
  function offload_model_from_device_for_memory_preservation (line 100) | def offload_model_from_device_for_memory_preservation(model, target_devi...
  function unload_complete_models (line 116) | def unload_complete_models(*args):
  function load_model_as_complete (line 126) | def load_model_as_complete(model, target_device, unload=True):

FILE: diffusers_helper/models/hunyuan_video_packed.py
  function pad_for_3d_conv (line 64) | def pad_for_3d_conv(x, kernel_size):
  function center_down_sample_3d (line 73) | def center_down_sample_3d(x, kernel_size):
  function get_cu_seqlens (line 82) | def get_cu_seqlens(text_mask, img_len):
  function apply_rotary_emb_transposed (line 99) | def apply_rotary_emb_transposed(x, freqs_cis):
  function attn_varlen_func (line 108) | def attn_varlen_func(q, k, v, cu_seqlens_q, cu_seqlens_kv, max_seqlen_q,...
  class HunyuanAttnProcessorFlashAttnDouble (line 143) | class HunyuanAttnProcessorFlashAttnDouble:
    method __call__ (line 144) | def __call__(self, attn, hidden_states, encoder_hidden_states, attenti...
  class HunyuanAttnProcessorFlashAttnSingle (line 189) | class HunyuanAttnProcessorFlashAttnSingle:
    method __call__ (line 190) | def __call__(self, attn, hidden_states, encoder_hidden_states, attenti...
  class CombinedTimestepGuidanceTextProjEmbeddings (line 219) | class CombinedTimestepGuidanceTextProjEmbeddings(nn.Module):
    method __init__ (line 220) | def __init__(self, embedding_dim, pooled_projection_dim):
    method forward (line 228) | def forward(self, timestep, guidance, pooled_projection):
  class CombinedTimestepTextProjEmbeddings (line 243) | class CombinedTimestepTextProjEmbeddings(nn.Module):
    method __init__ (line 244) | def __init__(self, embedding_dim, pooled_projection_dim):
    method forward (line 251) | def forward(self, timestep, pooled_projection):
  class HunyuanVideoAdaNorm (line 262) | class HunyuanVideoAdaNorm(nn.Module):
    method __init__ (line 263) | def __init__(self, in_features: int, out_features: Optional[int] = Non...
    method forward (line 270) | def forward(
  class HunyuanVideoIndividualTokenRefinerBlock (line 279) | class HunyuanVideoIndividualTokenRefinerBlock(nn.Module):
    method __init__ (line 280) | def __init__(
    method forward (line 306) | def forward(
  class HunyuanVideoIndividualTokenRefiner (line 329) | class HunyuanVideoIndividualTokenRefiner(nn.Module):
    method __init__ (line 330) | def __init__(
    method forward (line 354) | def forward(
  class HunyuanVideoTokenRefiner (line 376) | class HunyuanVideoTokenRefiner(nn.Module):
    method __init__ (line 377) | def __init__(
    method forward (line 404) | def forward(
  class HunyuanVideoRotaryPosEmbed (line 425) | class HunyuanVideoRotaryPosEmbed(nn.Module):
    method __init__ (line 426) | def __init__(self, rope_dim, theta):
    method get_frequency (line 432) | def get_frequency(self, dim, pos):
    method forward_inner (line 439) | def forward_inner(self, frame_indices, height, width, device):
    method forward (line 456) | def forward(self, frame_indices, height, width, device):
  class AdaLayerNormZero (line 463) | class AdaLayerNormZero(nn.Module):
    method __init__ (line 464) | def __init__(self, embedding_dim: int, norm_type="layer_norm", bias=Tr...
    method forward (line 473) | def forward(
  class AdaLayerNormZeroSingle (line 485) | class AdaLayerNormZeroSingle(nn.Module):
    method __init__ (line 486) | def __init__(self, embedding_dim: int, norm_type="layer_norm", bias=Tr...
    method forward (line 496) | def forward(
  class AdaLayerNormContinuous (line 508) | class AdaLayerNormContinuous(nn.Module):
    method __init__ (line 509) | def __init__(
    method forward (line 526) | def forward(self, x: torch.Tensor, emb: torch.Tensor) -> torch.Tensor:
  class HunyuanVideoSingleTransformerBlock (line 534) | class HunyuanVideoSingleTransformerBlock(nn.Module):
    method __init__ (line 535) | def __init__(
    method forward (line 565) | def forward(
  class HunyuanVideoTransformerBlock (line 608) | class HunyuanVideoTransformerBlock(nn.Module):
    method __init__ (line 609) | def __init__(
    method forward (line 643) | def forward(
  class ClipVisionProjection (line 683) | class ClipVisionProjection(nn.Module):
    method __init__ (line 684) | def __init__(self, in_channels, out_channels):
    method forward (line 689) | def forward(self, x):
  class HunyuanVideoPatchEmbed (line 694) | class HunyuanVideoPatchEmbed(nn.Module):
    method __init__ (line 695) | def __init__(self, patch_size, in_chans, embed_dim):
  class HunyuanVideoPatchEmbedForCleanLatents (line 700) | class HunyuanVideoPatchEmbedForCleanLatents(nn.Module):
    method __init__ (line 701) | def __init__(self, inner_dim):
    method initialize_weight_from_another_conv3d (line 708) | def initialize_weight_from_another_conv3d(self, another_layer):
  class HunyuanVideoTransformer3DModelPacked (line 727) | class HunyuanVideoTransformer3DModelPacked(ModelMixin, ConfigMixin, Peft...
    method __init__ (line 729) | def __init__(
    method install_image_projection (line 805) | def install_image_projection(self, in_channels):
    method install_clean_x_embedder (line 810) | def install_clean_x_embedder(self):
    method enable_gradient_checkpointing (line 814) | def enable_gradient_checkpointing(self):
    method disable_gradient_checkpointing (line 818) | def disable_gradient_checkpointing(self):
    method initialize_teacache (line 822) | def initialize_teacache(self, enable_teacache=True, num_steps=25, rel_...
    method gradient_checkpointing_method (line 832) | def gradient_checkpointing_method(self, block, *args):
    method process_input_hidden_states (line 839) | def process_input_hidden_states(
    method forward (line 898) | def forward(

FILE: diffusers_helper/pipelines/k_diffusion_hunyuan.py
  function flux_time_shift (line 9) | def flux_time_shift(t, mu=1.15, sigma=1.0):
  function calculate_flux_mu (line 13) | def calculate_flux_mu(context_length, x1=256, y1=0.5, x2=4096, y2=1.15, ...
  function get_flux_sigmas_from_mu (line 21) | def get_flux_sigmas_from_mu(n, mu):
  function sample_hunyuan (line 28) | def sample_hunyuan(

FILE: diffusers_helper/thread_utils.py
  class Listener (line 6) | class Listener:
    method _process_tasks (line 12) | def _process_tasks(cls):
    method add_task (line 30) | def add_task(cls, func, *args, **kwargs):
  function async_run (line 39) | def async_run(func, *args, **kwargs):
  class FIFOQueue (line 43) | class FIFOQueue:
    method __init__ (line 44) | def __init__(self):
    method push (line 48) | def push(self, item):
    method pop (line 52) | def pop(self):
    method top (line 58) | def top(self):
    method next (line 64) | def next(self):
  class AsyncStream (line 73) | class AsyncStream:
    method __init__ (line 74) | def __init__(self):

FILE: diffusers_helper/utils.py
  function min_resize (line 16) | def min_resize(x, m):
  function d_resize (line 33) | def d_resize(x, y):
  function resize_and_center_crop (line 45) | def resize_and_center_crop(image, target_width, target_height):
  function resize_and_center_crop_pytorch (line 63) | def resize_and_center_crop_pytorch(image, target_width, target_height):
  function resize_without_crop (line 82) | def resize_without_crop(image, target_width, target_height):
  function just_crop (line 91) | def just_crop(image, w, h):
  function write_to_json (line 105) | def write_to_json(data, file_path):
  function read_from_json (line 113) | def read_from_json(file_path):
  function get_active_parameters (line 119) | def get_active_parameters(m):
  function cast_training_params (line 123) | def cast_training_params(m, dtype=torch.float32):
  function separate_lora_AB (line 132) | def separate_lora_AB(parameters, B_patterns=None):
  function set_attr_recursive (line 148) | def set_attr_recursive(obj, attr, value):
  function print_tensor_list_size (line 156) | def print_tensor_list_size(tensors):
  function batch_mixture (line 177) | def batch_mixture(a, b=None, probability_a=0.5, mask_a=None):
  function zero_module (line 193) | def zero_module(module):
  function supress_lower_channels (line 200) | def supress_lower_channels(m, k, alpha=0.01):
  function freeze_module (line 210) | def freeze_module(m):
  function get_latest_safetensors (line 218) | def get_latest_safetensors(folder_path):
  function generate_random_prompt_from_tags (line 229) | def generate_random_prompt_from_tags(tags_str, min_length=3, max_length=...
  function interpolate_numbers (line 236) | def interpolate_numbers(a, b, n, round_to_int=False, gamma=1.0):
  function uniform_random_by_intervals (line 243) | def uniform_random_by_intervals(inclusive, exclusive, n, round_to_int=Fa...
  function soft_append_bcthw (line 252) | def soft_append_bcthw(history, current, overlap=0):
  function save_bcthw_as_mp4 (line 266) | def save_bcthw_as_mp4(x, output_filename, fps=10, crf=0):
  function save_bcthw_as_png (line 283) | def save_bcthw_as_png(x, output_filename):
  function save_bchw_as_png (line 292) | def save_bchw_as_png(x, output_filename):
  function add_tensors_with_padding (line 301) | def add_tensors_with_padding(tensor1, tensor2):
  function print_free_mem (line 320) | def print_free_mem():
  function print_gpu_parameters (line 330) | def print_gpu_parameters(device, state_dict, log_count=1):
  function visualize_txt_as_img (line 345) | def visualize_txt_as_img(width, height, text, font_path='font/DejaVuSans...
  function blue_mark (line 383) | def blue_mark(x):
  function green_mark (line 391) | def green_mark(x):
  function frame_mark (line 398) | def frame_mark(x):
  function pytorch2numpy (line 408) | def pytorch2numpy(imgs):
  function numpy2pytorch (line 419) | def numpy2pytorch(imgs):
  function duplicate_prefix_to_suffix (line 426) | def duplicate_prefix_to_suffix(x, count, zero_out=False):
  function weighted_mse (line 433) | def weighted_mse(a, b, weight):
  function clamped_linear_interpolation (line 437) | def clamped_linear_interpolation(x, x_min, y_min, x_max, y_max, sigma=1.0):
  function expand_to_dims (line 444) | def expand_to_dims(x, target_dims):
  function repeat_to_batch_size (line 448) | def repeat_to_batch_size(tensor: torch.Tensor, batch_size: int):
  function dim5 (line 465) | def dim5(x):
  function dim4 (line 469) | def dim4(x):
  function dim3 (line 473) | def dim3(x):
  function crop_or_pad_yield_mask (line 477) | def crop_or_pad_yield_mask(x, length):
  function extend_dim (line 492) | def extend_dim(x, dim, minimal_length, zero_pad=False):
  function lazy_positional_encoding (line 510) | def lazy_positional_encoding(t, repeats=None):
  function state_dict_offset_merge (line 527) | def state_dict_offset_merge(A, B, C=None):
  function state_dict_weighted_merge (line 544) | def state_dict_weighted_merge(state_dicts, weights):
  function group_files_by_folder (line 571) | def group_files_by_folder(all_files):
  function generate_timestamp (line 584) | def generate_timestamp():
  function write_PIL_image_with_png_info (line 592) | def write_PIL_image_with_png_info(image, metadata, path):
  function torch_safe_save (line 603) | def torch_safe_save(content, path):
  function move_optimizer_to_device (line 609) | def move_optimizer_to_device(optimizer, device):