Functions — github.com/ali-vilab/ACE

hub / github.com/ali-vilab/ACE_plus / functions

Functions190 in github.com/ali-vilab/ACE_plus

⨍ Functions 190◇ Types & classes 38

↓ 14 callersMethodload

(self, module)

inference/registry.py:91

↓ 12 callersMethod__init__

(self, dim: int)

modules/layers.py:154

↓ 8 callersMethodkeys

(self)

model_convert.py:33

↓ 5 callersFunctionattention

(q: Tensor, k: Tensor, v: Tensor, pe: Tensor, mask: Tensor | None = None, backend = 'pytorch')

modules/layers.py:24

↓ 5 callersMethodencode

(self, text, return_mask = False)

modules/embedder.py:110

↓ 5 callersMethodunload

(self, module)

inference/registry.py:118

↓ 4 callersFunctionedit_preprocess

(processor, device, edit_image, edit_mask)

inference/utils.py:13

↓ 4 callersMethodencode_first_stage

(self, x, **kwargs)

modules/ace_plus_ldm.py:101

↓ 4 callersMethodencode_list

(self, text_list, return_mask=True)

modules/embedder.py:132

↓ 4 callersFunctionensure_limit_sequence

(image, max_seq_len = 4096, d = 16, interpolation=InterpolationMode.BILINEAR)

modules/ace_plus_dataset.py:64

↓ 4 callersFunctionload_image

(prefix, img_path, cvt_type=None)

modules/ace_plus_dataset.py:16

↓ 4 callersMethodreshape_func

(self, mask)

modules/ace_plus_ldm.py:210

↓ 4 callersMethodsave_results

(self, results)

modules/ace_plus_solver.py:91

↓ 4 callersFunctiontimestep_embedding

Create sinusoidal timestep embeddings. :param t: a 1-D Tensor of N indices, one per batch element. These may be fractio

modules/layers.py:106

↓ 3 callersMethodafter_iter

(self, solver)

modules/checkpoint.py:86

↓ 3 callersMethodinit_from_cfg

(self, cfg)

inference/ace_plus_diffusers.py:28

↓ 3 callersFunctiontransform_mask

(mask)

modules/ace_plus_dataset.py:27

↓ 2 callersMethod__init__

(self, cfg, logger=None)

modules/flux.py:86

↓ 2 callersMethodencode

(self, positive, negative, vae, ucpixels,

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:81

↓ 2 callersMethodencode_list_of_list

(self, text_list, return_mask=True)

modules/embedder.py:144

↓ 2 callersFunctionensure_src_align_target_h_mode

(src_image, size, image_id, interpolation=InterpolationMode.BILINEAR)

modules/ace_plus_dataset.py:30

↓ 2 callersFunctionensure_src_align_target_padding_mode

(src_image, size, image_id, size_h = [], interpolation=InterpolationMode.BILINEAR)

modules/ace_plus_dataset.py:40

↓ 2 callersMethodforward

(self, q: Tensor, k: Tensor, v: Tensor)

modules/layers.py:159

↓ 2 callersMethodgrad_clip

(self, optimizer)

modules/checkpoint.py:76

↓ 2 callersMethodimage_check

(self, image)

inference/utils.py:34

↓ 2 callersMethodimage_check

(self, image)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:255

↓ 2 callersMethodimage_check

(self, image)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:462

↓ 2 callersMethodnoise_sample

(self, num_samples, h, w, seed, dtype=torch.bfloat16)

modules/ace_plus_ldm.py:114

↓ 2 callersMethodparse_ref_and_edit

(self, src_image, modify_image, src_image_mask,

modules/ace_plus_ldm.py:129

↓ 2 callersMethodpostprocess

(self, image, slice_w, out_w, out_h)

inference/utils.py:125

↓ 2 callersMethodpreprocess

(self, reference_image=None, edit_image=None, edit_ma

inference/utils.py:46

↓ 2 callersFunctionrun_one_case

(pipe, input_image = None, input_mask = None, input_reference_

infer_fft.py:35

↓ 2 callersFunctionrun_one_case

(pipe, input_image = None, input_mask = None, input_reference_

infer_lora.py:30

↓ 2 callersFunctiontransform_image

(image, std = 0.5, mean = 0.5)

modules/ace_plus_dataset.py:25

↓ 2 callersMethodunpack

(self, x: Tensor, height: int, width: int)

modules/flux.py:162

↓ 1 callersMethod__init__

(self, cfg, logger=None)

modules/embedder.py:61

↓ 1 callersMethod_clean

(self, text)

modules/embedder.py:156

↓ 1 callersMethod_convert_float8

(byte_tensor, dtype_str, shape)

model_convert.py:97

↓ 1 callersMethod_deserialize_tensor

(self, tensor_bytes, metadata)

model_convert.py:58

↓ 1 callersMethod_get

(self, index)

modules/ace_plus_dataset.py:160

↓ 1 callersMethod_get_torch_dtype

(dtype_str)

model_convert.py:76

↓ 1 callersMethod_read_header

(self)

model_convert.py:52

↓ 1 callersFunctionapply_rope

(xq: Tensor, xk: Tensor, freqs_cis: Tensor)

modules/layers.py:82

↓ 1 callersMethodconstruct_edit_image

(self, edit_image, edit_mask)

demo_fft.py:90

↓ 1 callersMethodconstruct_edit_image

(self, edit_image, edit_mask)

demo_lora.py:103

↓ 1 callersFunctionconvert_to_comfyui_lora

(ori_sd, prefix = "lora_unet")

modules/checkpoint.py:16

↓ 1 callersMethodcreate_ui

(self)

demo_fft.py:110

↓ 1 callersMethodcreate_ui

(self)

demo_lora.py:126

↓ 1 callersMethoddecode_first_stage

(self, z)

modules/ace_plus_ldm.py:112

↓ 1 callersMethodeasy_lora_merge

(self, ori_sd, lora_sd, scale = 1.0)

modules/flux.py:343

↓ 1 callersMethodedit_preprocess

(self, processor, device, edit_image, edit_mask)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:274

↓ 1 callersMethodedit_preprocess

(self, processor, device, edit_image, edit_mask)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:481

↓ 1 callersMethodforward_editing

(self, src_image_list=[], modify_image_list=None,

modules/ace_plus_ldm.py:341

↓ 1 callersMethodforward_train

Args: src_image: list of list of src_image src_image_mask: list of list of src_image_mask ima

modules/ace_plus_ldm.py:221

↓ 1 callersMethodget_tensor

(self, key)

model_convert.py:36

↓ 1 callersMethodinit_from_ckpt

(self, path, model, ignore_keys=list())

inference/registry.py:61

↓ 1 callersMethodload_default

(self, cfg)

inference/ace_plus_diffusers.py:22

↓ 1 callersMethodload_yaml

(self, cfg_file)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:250

↓ 1 callersMethodload_yaml

(self, cfg_file)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:457

↓ 1 callersFunctionmem_eff_save_file

(tensors: Dict[str, torch.Tensor], filename: str, metadata: Dict[str, Any] = None)

model_convert.py:109

↓ 1 callersMethodmerge_blackforest_lora

(self, ori_sd, lora_sd, scale = 1.0)

modules/flux.py:284

↓ 1 callersMethodmerge_comfyui_lora

(self, ori_sd, lora_sd, scale = 1.0)

modules/flux.py:320

↓ 1 callersMethodmerge_diffuser_lora

(self, ori_sd, lora_sd, scale=1.0)

modules/flux.py:172

↓ 1 callersMethodmerge_swift_lora

(self, ori_sd, lora_sd, scale = 1.0)

modules/flux.py:262

↓ 1 callersMethodprepare_input

(self, x, context, y, x_shape=None)

modules/flux.py:151

↓ 1 callersMethodprepare_input

(self, x, cond)

modules/flux.py:523

↓ 1 callersMethodprepare_input

(self, image, mask, batch_size=1,

inference/ace_plus_diffusers.py:41

↓ 1 callersMethodread_data_list

(self, delimiter, fields, prefix,

modules/ace_plus_dataset.py:123

↓ 1 callersFunctionread_safetensors_metadata

(path: str)

model_convert.py:175

↓ 1 callersFunctionrope

(pos: Tensor, dim: int, theta: int)

modules/layers.py:72

↓ 1 callersFunctionrun

()

infer_fft.py:82

↓ 1 callersFunctionrun

()

infer_lora.py:73

↓ 1 callersMethodrun_eval

(self)

modules/ace_plus_solver.py:19

↓ 1 callersFunctionrun_gr

(cfg)

demo_fft.py:526

↓ 1 callersFunctionrun_gr

(cfg)

demo_lora.py:504

↓ 1 callersMethodset_callbacks

(self, *args, **kwargs)

demo_fft.py:271

↓ 1 callersMethodset_callbacks

(self, *args, **kwargs)

demo_lora.py:258

↓ 1 callersMethodtrans_pil_tensor

(self, pil_image)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:267

↓ 1 callersMethodtrans_pil_tensor

(self, pil_image)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:474

↓ 1 callersMethodunpack

(self, x: Tensor, cond: dict = None, x_seq_length: list = None)

modules/flux.py:563

↓ 1 callersFunctionupdate_config

(cfg)

run_train.py:35

↓ 1 callersFunctionvalidate_metadata

(metadata: Dict[str, Any])

model_convert.py:126

MethodINPUT_TYPES

(s)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:38

MethodINPUT_TYPES

(s)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:61

MethodINPUT_TYPES

(s)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:129

MethodINPUT_TYPES

(s)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:207

MethodINPUT_TYPES

(s)

workflow/ComfyUI-ACE_Plus/ace_plus_fft_node.py:420

Method__call__

(self, reference_image=None, edit_image=None, edit_mask=Non

inference/ace_plus_diffusers.py:69

Method__call__

(self, reference_image=None, edit_image=None, edit_mask=Non

inference/ace_plus_inference.py:27

Method__call__

(self, input, **kwargs)

inference/registry.py:210

(self, exc_type, exc_val, exc_tb)

model_convert.py:30

Method__getitem__

(self, index)

modules/ace_plus_dataset.py:156

Method__init__

(self, infer_dir = "./config/ace_plus_fft.yaml" )

demo_fft.py:54

Method__init__

(self, infer_dir = "./config/ace_plus_diffusers_infer.yaml", model_list='./m

(self, dim: int, theta: int, axes_dim: list[int])

modules/layers.py:90

Method__init__

(self, in_dim: int, hidden_dim: int)

modules/layers.py:131

Method__init__

(self, dim: int)

modules/layers.py:142

Method__init__

(self, dim: int, num_heads: int = 8, qkv_bias: bool = False)

modules/layers.py:166

next →1–100 of 190, ranked by callers