Functions — github.com/aistudynow/Comfyui-bitdance

hub / github.com/aistudynow/Comfyui-bitdance / functions

Functions163 in github.com/aistudynow/Comfyui-bitdance

⨍ Functions 163◇ Types & classes 35

↓ 9 callersFunction_empty_cuda_cache

()

nodes.py:524

↓ 6 callersMethod__init__

( self, ch_target, ch_cond, ch_latent, depth_latent, depth_ada

bitdance_arch/vision_head/flow_head_parallel_x.py:44

↓ 6 callersMethod__init__

(self, *, ch, out_ch, in_channels, num_res_blocks, z_channels, ch_mult=(1, 2, 2, 4), resoluti

bitdance_arch/vision_encoder/autoencoder.py:60

↓ 6 callersFunction_get_filename_list_multi

(folder_types: Tuple[str, ...])

nodes.py:296

↓ 6 callersFunction_is_scale_key

(key: str)

nodes.py:377

↓ 6 callersFunction_named_device

(name: str)

nodes.py:1187

↓ 5 callersFunctionswish

(x)

bitdance_arch/vision_encoder/autoencoder.py:10

↓ 4 callersFunction_apply_manual_cast_hint

(module: torch.nn.Module)

nodes.py:514

↓ 4 callersFunction_cache_seq_len

(past_key_values)

nodes.py:1350

↓ 4 callersFunction_find_first_existing

(root: Path, filenames: Iterable[str])

nodes.py:359

↓ 4 callersFunction_get_full_path_from_folder_types

(folder_types: Tuple[str, ...], filename: str)

nodes.py:307

↓ 4 callersFunction_is_aux_quant_key

(key: str)

nodes.py:387

↓ 4 callersFunction_read_json

(path: Path)

local_model.py:110

↓ 4 callersFunction_token_id

(tokenizer, token: str)

nodes.py:1343

↓ 4 callersMethodload

( self, model_name: str, text_encoder_name: str, vae_name: str, q

nodes.py:1883

↓ 4 callersFunctionset_bitdance_attention_mode

(mode: str = "auto")

bitdance_arch/vision_head/flow_head_parallel_x.py:18

↓ 3 callersFunction_bitdance_resolution_label

(height: int, width: int)

nodes.py:74

↓ 3 callersFunction_dequantize_state_dict

( state_dict: Mapping[str, torch.Tensor], *, external_scales: Optional[Mapping[str, torch.Tenso

nodes.py:437

↓ 3 callersFunction_extract_prefixed_substate

( raw_sd: Mapping[str, torch.Tensor], prefixes: Tuple[str, ...] )

nodes.py:980

↓ 3 callersFunction_load_component_state_dict

( module: torch.nn.Module, file_path: Path, target_dtype: torch.dtype, *, strict: boo

nodes.py:529

↓ 3 callersFunction_load_component_state_dict_from_raw

( module: torch.nn.Module, raw_sd: Mapping[str, torch.Tensor], target_dtype: torch.dtype,

nodes.py:546

↓ 3 callersFunction_load_state_dict

(module: torch.nn.Module, state_dict: Dict[str, torch.Tensor], strict: bool)

nodes.py:506

↓ 3 callersFunction_normalize_text_attention_mode

(mode: str)

local_model.py:138

↓ 3 callersFunction_resolve_dtype

(dtype_name: str)

nodes.py:271

↓ 3 callersFunction_scale_key_candidates

(weight_key: str)

nodes.py:391

↓ 3 callersFunction_try_resolve_model_root_from_path

(file_path: Path)

nodes.py:352

↓ 3 callersMethodencode

(self, vae: BitDanceVAERuntime, image: torch.Tensor, parallel_num: int)

nodes.py:2363

↓ 3 callersMethodencode

(self, x)

bitdance_arch/vision_encoder/autoencoder.py:385

↓ 3 callersFunctionstrip_diffusers_metadata

(config: Dict[str, Any])

local_model.py:106

↓ 2 callersMethod__init__

(self, runtime: BitDanceVAERuntime)

nodes.py:264

↓ 2 callersFunction_configure_text_model_runtime

( model: torch.nn.Module, config: Any, *, attention_mode: str = "auto", rms_norm_function:

local_model.py:181

↓ 2 callersFunction_encode_bitdance_text_prompts

( text_encoder: BitDanceTextRuntime, positive_prompt: str, negative_prompt: str, *, d

nodes.py:1265

↓ 2 callersFunction_get_base_llm

(llm_model: torch.nn.Module)

nodes.py:1253

↓ 2 callersFunction_get_embed_tokens

(base_llm: torch.nn.Module, llm_model: torch.nn.Module)

nodes.py:1259

↓ 2 callersFunction_load_cached_text_embed

(cache_key: str)

nodes.py:1108

↓ 2 callersFunction_load_tokenizer_for_single_file

(text_encoder_file: Path)

nodes.py:1078

↓ 2 callersFunction_resolve_model_root_from_path

(file_path: Path)

nodes.py:323

↓ 2 callersFunction_save_cached_text_embed

(cache_key: str, tensor: torch.Tensor)

nodes.py:1125

↓ 2 callersFunction_set_param_or_buffer_by_key

( root: torch.nn.Module, modules_map: Dict[str, torch.nn.Module], key: str, tensor: torch.

nodes.py:729

↓ 2 callersFunction_strip_first_matching_prefix

(key: str, prefixes: Tuple[str, ...])

nodes.py:562

↓ 2 callersFunction_text_embed_cache_key

(kind: str, prompt: str, hidden_size: int)

nodes.py:1099

↓ 2 callersFunction_text_embed_cache_path

(cache_key: str)

nodes.py:1104

↓ 2 callersFunction_try_import_init_empty_weights

()

nodes.py:678

↓ 2 callersMethoddecode

(self, vae: BitDanceVAERuntime, bitdance_latent: BitDanceLatentRuntime)

nodes.py:2269

↓ 2 callersFunctioneuler_step

(x, v, dt: float, cfg: float, cfg_mult: int)

bitdance_arch/vision_head/sampling_x.py:24

↓ 2 callersFunctionget_velocity_from_cfg

(velocity, cfg, cfg_mult)

bitdance_arch/vision_head/sampling_x.py:16

↓ 2 callersMethodset_weight

(self, weight: torch.Tensor, weight_scale: Optional[torch.Tensor] = None)

nodes.py:181

↓ 2 callersFunctiontime_shift_func

(t: torch.Tensor, flow_shift: float = 1., sigma: float = 1.)

bitdance_arch/vision_head/flow_head_parallel_x.py:40

↓ 1 callersFunction_apply_loader_device_selection

( model_runtime: BitDanceModelRuntime, text_runtime: BitDanceTextRuntime, vae_runtime: BitDance

nodes.py:1193

↓ 1 callersFunction_bitdance_prompt_templates

(positive_prompt: str, negative_prompt: str)

nodes.py:1090

↓ 1 callersFunction_build_model_runtime_from_single_file

(main_file: Path, target_dtype: torch.dtype)

nodes.py:1388

↓ 1 callersFunction_build_pos_embed_1d

(hidden_size: int, vae_patch_size: int, device: torch.device)

nodes.py:1360

↓ 1 callersFunction_build_text_runtime_from_single_file

( text_file: Path, target_dtype: torch.dtype, *, attention_mode: str = "auto", rms_no

nodes.py:1423

↓ 1 callersFunction_build_text_runtime_from_single_file_fp8

( text_file: Path, *, target_dtype: torch.dtype, attention_mode: str = "auto", rms_no

nodes.py:894

↓ 1 callersFunction_build_vae_runtime_from_single_file

(vae_file: Path, target_dtype: torch.dtype)

nodes.py:1510

↓ 1 callersFunction_coerce_bitdance_resolution

(resolution: Any)

nodes.py:1931

↓ 1 callersMethod_dequant_weight

(self, device: torch.device, dtype: torch.dtype)

nodes.py:195

↓ 1 callersFunction_detect_diffusers_format

(state_dict: Mapping[str, torch.Tensor])

nodes.py:428

↓ 1 callersFunction_download_tokenizer_source

()

nodes.py:1051

↓ 1 callersFunction_find_tokenizer_source_near

(path: Path)

nodes.py:1025

↓ 1 callersFunction_get_2d_embed

( pos_embed_1d: torch.Tensor, hidden_size: int, h: int, w: int, ps: int, )

nodes.py:1373

↓ 1 callersFunction_get_checkpoint_path

(ckpt_name: str)

nodes.py:285

↓ 1 callersFunction_load_json

(path: Path)

nodes.py:367

↓ 1 callersFunction_load_llm_shards

( model: torch.nn.Module, model_root: Path, target_dtype: torch.dtype, )

nodes.py:1221

↓ 1 callersFunction_move_bitdance_model_runtime_to_offload

(model_runtime: Optional[BitDanceModelRuntime])

nodes.py:1137

↓ 1 callersFunction_parse_bitdance_resolution_label

(label: str)

nodes.py:81

↓ 1 callersFunction_rename_diffusers_to_comfy_key

(key: str)

nodes.py:402

↓ 1 callersFunction_replace_qwen_linears_with_fp8

(model: torch.nn.Module, *, skip_lm_head: bool = True)

nodes.py:702

↓ 1 callersFunction_replace_rmsnorm_with_pytorch

(module: nn.Module)

local_model.py:152

↓ 1 callersFunction_resolve_model_root_from_checkpoint

(ckpt_name: str)

nodes.py:341

↓ 1 callersFunction_safe_get_filename_list

(folder_type: str)

nodes.py:289

↓ 1 callersFunction_set_child_module

(root: torch.nn.Module, module_name: str, new_module: torch.nn.Module)

nodes.py:694

↓ 1 callersFunction_split_main_model_state_dict

(raw_sd: Mapping[str, torch.Tensor])

nodes.py:992

↓ 1 callersFunction_stream_load_qwen_text_encoder_fp8

( llm_model: torch.nn.Module, text_file: Path, *, target_dtype: torch.dtype, target_d

nodes.py:795

↓ 1 callersFunction_stream_load_safetensors_into_module

Streaming loader to avoid materializing a huge CPU state_dict in RAM. This is especially important for merged BitDance text encoder checkpo

nodes.py:569

↓ 1 callersMethoddecode

(self, quant)

bitdance_arch/vision_encoder/autoencoder.py:514

↓ 1 callersFunctiondownload

Minimal local-path compatible replacement for BitDance's fs.download. ComfyUI nodes here load safetensors directly and do not require hdfs su

bitdance_arch/utils_fs.py:7

↓ 1 callersFunctioneuler

( input_dim, forward_fn, c, cfg: float = 1.0, num_sampling_steps: int = 50, progress_c

bitdance_arch/vision_head/sampling_x.py:106

↓ 1 callersFunctioneuler_maruyama

( input_dim, forward_fn, c: torch.Tensor, cfg: float = 1.0, num_sampling_steps: int = 20,

bitdance_arch/vision_head/sampling_x.py:44

↓ 1 callersFunctioneuler_maruyama_step

(x, v, t, dt: float, cfg: float, cfg_mult: int)

bitdance_arch/vision_head/sampling_x.py:33

↓ 1 callersFunctionget_score_from_velocity

(velocity, x, t)

bitdance_arch/vision_head/sampling_x.py:6

↓ 1 callersMethodinitialize_weights

(self)

bitdance_arch/vision_head/flow_head_parallel_x.py:341

↓ 1 callersMethodload_from_ckpt

(self, checkpoint)

bitdance_arch/vision_encoder/autoencoder.py:368

↓ 1 callersMethodload_model

(self, ckpt_name: str, dtype: str, attention_mode: str = "auto")

nodes.py:1544

↓ 1 callersMethodload_text_encoder

( self, ckpt_name: str, dtype: str, attention_mode: str = "auto",

nodes.py:1608

↓ 1 callersMethodload_vae

(self, ckpt_name: str, dtype: str)

nodes.py:1813

↓ 1 callersMethodset_bias

(self, bias: torch.Tensor)

nodes.py:191

↓ 1 callersFunctiontime_shift_func

(t: torch.Tensor, flow_shift: float = 1., sigma: float = 1.)

bitdance_arch/vision_head/sampling_x.py:3

↓ 1 callersFunctiontimestep_embedding

(t, dim, max_period=10000, time_factor: float = 1000.0)

bitdance_arch/vision_head/flow_head_parallel_x.py:23

( self, in_features: int, out_features: int, bias: bool = True, *

nodes.py:155

next →1–100 of 163, ranked by callers