MCPcopy Create free account

hub / github.com/TencentARC/FluxKits / functions

Functions314 in github.com/TencentARC/FluxKits

⨍ Functions 314◇ Types & classes 86
↓ 15 callersMethod__init__
(self, dim: int)
flux-mini/src/flux/modules/layers.py:75
↓ 15 callersMethod__init__
(self, dim: int)
flux-npu/src/flux/modules/layers.py:75
↓ 8 callersFunctionattention
(q, k, v, pe)
flux-mini/src/flux/math.py:4
↓ 8 callersFunctionattention
(q: Tensor, k: Tensor, v: Tensor, pe: Tensor)
flux-npu/src/flux/math.py:7
↓ 7 callersMethod__init__
( self, resolution: int, in_channels: int, ch: int, ch_mult: list,
flux-mini/src/flux/modules/autoencoder.py:110
↓ 7 callersMethod__init__
( self, resolution: int, in_channels: int, ch: int, ch_mult: list[int]
flux-npu/src/flux/modules/autoencoder.py:110
↓ 7 callersFunctionprepare
(t5: HFEmbedder, clip: HFEmbedder, img: Tensor, prompt: str)
flux-mini/src/flux/sampling.py:32
↓ 7 callersFunctionprepare
(t5: HFEmbedder, clip: HFEmbedder, img: Tensor, prompt: str)
flux-npu/src/flux/sampling.py:32
↓ 5 callersMethoddecode
(self, z: Tensor)
flux-mini/src/flux/modules/autoencoder.py:307
↓ 5 callersMethoddecode
(self, z: Tensor)
flux-npu/src/flux/modules/autoencoder.py:307
↓ 5 callersMethodencode
(self, x: Tensor)
flux-mini/src/flux/modules/autoencoder.py:302
↓ 5 callersMethodset_processor
(self, processor)
flux-mini/src/flux/modules/layers.py:372
↓ 5 callersMethodset_processor
(self, processor)
flux-npu/src/flux/modules/layers.py:372
↓ 4 callersMethodencode
(self, x: Tensor)
flux-npu/src/flux/modules/autoencoder.py:302
↓ 4 callersFunctionget_noise
( num_samples: int, height: int, width: int, device: torch.device, dtype: torch.dtype,
flux-mini/src/flux/sampling.py:12
↓ 4 callersFunctionget_noise
( num_samples: int, height: int, width: int, device: torch.device, dtype: torch.dtype,
flux-npu/src/flux/sampling.py:12
↓ 4 callersFunctionget_schedule
( num_steps: int, image_seq_len: int, base_shift: float = 0.5, max_shift: float = 1.15, sh
flux-mini/src/flux/sampling.py:78
↓ 4 callersFunctionget_schedule
( num_steps: int, image_seq_len: int, base_shift: float = 0.5, max_shift: float = 1.15, sh
flux-npu/src/flux/sampling.py:78
↓ 4 callersFunctionload_ae
(name: str, device: str = "cpu", hf_download: bool = True)
flux-mini/src/flux/util.py:411
↓ 4 callersFunctionload_ae
(name: str, device: str = "npu", hf_download: bool = True)
flux-npu/src/flux/util.py:411
↓ 4 callersFunctionload_checkpoint
(local_path, repo_id, name)
flux-mini/src/flux/util.py:40
↓ 4 callersFunctionload_checkpoint
(local_path, repo_id, name)
flux-npu/src/flux/util.py:40
↓ 4 callersFunctionload_clip
(device: str = "cpu")
flux-mini/src/flux/util.py:407
↓ 4 callersFunctionload_clip
(device: str = "npu")
flux-npu/src/flux/util.py:407
↓ 4 callersFunctionload_t5
(device: str = "cpu", max_length: int = 512)
flux-mini/src/flux/util.py:402
↓ 4 callersFunctionload_t5
(device: str = "npu", max_length: int = 512)
flux-npu/src/flux/util.py:402
↓ 4 callersFunctionprint_load_warning
(missing: list[str], unexpected: list[str])
flux-mini/src/flux/util.py:300
↓ 4 callersFunctionprint_load_warning
(missing: list[str], unexpected: list[str])
flux-npu/src/flux/util.py:300
↓ 4 callersMethodsave
Save the generated image to a local path
flux-mini/src/flux/api.py:179
↓ 4 callersMethodsave
Save the generated image to a local path
flux-npu/src/flux/api.py:179
↓ 4 callersFunctionswish
(x: Tensor)
flux-mini/src/flux/modules/autoencoder.py:21
↓ 4 callersFunctionswish
(x: Tensor)
flux-npu/src/flux/modules/autoencoder.py:21
↓ 4 callersFunctiontimestep_embedding
Create sinusoidal timestep embeddings. :param t: a 1-D Tensor of N indices, one per batch element. These may be fractio
flux-mini/src/flux/modules/layers.py:28
↓ 4 callersFunctiontimestep_embedding
Create sinusoidal timestep embeddings. :param t: a 1-D Tensor of N indices, one per batch element. These may be fractio
flux-npu/src/flux/modules/layers.py:28
↓ 4 callersFunctionunpack
(x: Tensor, height: int, width: int)
flux-mini/src/flux/sampling.py:252
↓ 4 callersFunctionunpack
(x: Tensor, height: int, width: int)
flux-npu/src/flux/sampling.py:256
↓ 3 callersFunctiondenoise_simple
(model, img, img_ids, txt, txt_ids, vec, timesteps, guidance=4.0)
flux-mini/src/flux/sampling.py:153
↓ 3 callersFunctiondenoise_simple
(model, img, img_ids, txt, txt_ids, vec, timesteps, guidance=4.0)
flux-npu/src/flux/sampling.py:97
↓ 3 callersMethodoffload_model_to_cpu
(self, *models)
flux-mini/src/flux/xflux_pipeline.py:341
↓ 3 callersMethodoffload_model_to_cpu
(self, *models)
flux-npu/src/flux/xflux_pipeline.py:340
↓ 3 callersMethodset_attn_processor
r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o
flux-mini/src/flux/model.py:116
↓ 3 callersMethodset_attn_processor
r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o
flux-npu/src/flux/model.py:116
↓ 2 callersFunctionHWC3
(x)
flux-mini/src/flux/util.py:70
↓ 2 callersFunctionHWC3
(x)
flux-npu/src/flux/util.py:70
↓ 2 callersFunctionapp
()
flux-mini/src/flux/cli.py:247
↓ 2 callersFunctionapp
()
flux-npu/src/flux/cli.py:250
↓ 2 callersMethodget_image_proj
( self, image_prompt: Tensor, )
flux-mini/src/flux/xflux_pipeline.py:138
↓ 2 callersMethodget_image_proj
( self, image_prompt: Tensor, )
flux-npu/src/flux/xflux_pipeline.py:137
↓ 2 callersFunctionload_flow_model
(name: str, device: str = "cpu", hf_download: bool = True)
flux-mini/src/flux/util.py:315
↓ 2 callersFunctionload_flow_model
(name: str, device: str = "npu", hf_download: bool = True)
flux-npu/src/flux/util.py:315
↓ 2 callersFunctionload_flow_model2
(name: str, args, device: str = "cuda", hf_download: bool = True)
flux-mini/src/flux/util.py:335
↓ 2 callersFunctionload_flow_model2
(name: str, args, device: str = "cuda", hf_download: bool = True)
flux-npu/src/flux/util.py:335
↓ 2 callersFunctionloader
(train_batch_size, num_workers, **args)
flux-mini/image_datasets/dataset.py:91
↓ 2 callersFunctionloader
(train_batch_size, num_workers, **args)
flux-npu/image_datasets/dataset.py:91
↓ 2 callersFunctionpad64
(x)
flux-mini/src/flux/util.py:67
↓ 2 callersFunctionpad64
(x)
flux-npu/src/flux/util.py:67
↓ 2 callersFunctionparse_prompt
(options: SamplingOptions)
flux-mini/src/flux/cli.py:28
↓ 2 callersFunctionparse_prompt
(options: SamplingOptions)
flux-npu/src/flux/cli.py:30
↓ 2 callersMethodrequest
Request to generate the image.
flux-mini/src/flux/api.py:101
↓ 2 callersMethodrequest
Request to generate the image.
flux-npu/src/flux/api.py:101
↓ 2 callersFunctionsafer_memory
(x)
flux-mini/src/flux/util.py:88
↓ 2 callersFunctionsafer_memory
(x)
flux-npu/src/flux/util.py:88
↓ 2 callersMethodset_controlnet
(self, control_type: str, local_path: str = None, repo_id: str = None, name: str = None)
flux-mini/src/flux/xflux_pipeline.py:128
↓ 2 callersMethodset_controlnet
(self, control_type: str, local_path: str = None, repo_id: str = None, name: str = None)
flux-npu/src/flux/xflux_pipeline.py:127
↓ 2 callersMethodset_ip
(self, local_path: str = None, repo_id = None, name: str = None)
flux-mini/src/flux/xflux_pipeline.py:54
↓ 2 callersMethodset_ip
(self, local_path: str = None, repo_id = None, name: str = None)
flux-npu/src/flux/xflux_pipeline.py:53
↓ 2 callersMethodupdate_model_with_lora
(self, checkpoint, lora_weight)
flux-mini/src/flux/xflux_pipeline.py:108
↓ 2 callersMethodupdate_model_with_lora
(self, checkpoint, lora_weight)
flux-npu/src/flux/xflux_pipeline.py:107
↓ 2 callersFunctionzero_module
(module)
flux-mini/src/flux/controlnet.py:27
↓ 2 callersFunctionzero_module
(module)
flux-npu/src/flux/controlnet.py:26
↓ 1 callersMethod__repr__
(self)
flux-mini/src/flux/api.py:21
↓ 1 callersMethod__repr__
(self)
flux-npu/src/flux/api.py:21
↓ 1 callersFunctionapply_rope
(xq, xk, freqs_cis)
flux-mini/src/flux/math.py:23
↓ 1 callersFunctionapply_rope
(xq: Tensor, xk: Tensor, freqs_cis: Tensor)
flux-npu/src/flux/math.py:26
↓ 1 callersMethodattention
(self, h_: Tensor)
flux-mini/src/flux/modules/autoencoder.py:37
↓ 1 callersMethodattention
(self, h_: Tensor)
flux-npu/src/flux/modules/autoencoder.py:37
↓ 1 callersFunctioncrop_to_aspect_ratio
(image, ratio="16:9")
flux-mini/image_datasets/dataset.py:29
↓ 1 callersFunctioncrop_to_aspect_ratio
(image, ratio="16:9")
flux-npu/image_datasets/dataset.py:29
↓ 1 callersFunctiondenoise
( model: Flux, # model input img: Tensor, img_ids: Tensor, txt: Tensor, txt_ids: Tenso
flux-mini/src/flux/sampling.py:97
↓ 1 callersFunctiondenoise
( model: Flux, img: Tensor, img_ids: Tensor, txt: Tensor, txt_ids: Tensor, vec: Tensor
flux-npu/src/flux/sampling.py:119
↓ 1 callersFunctiondenoise_controlnet
(model: None, controlnet, # model input img: Tensor, img_ids: Tensor, txt: Tensor, txt
flux-mini/src/flux/sampling.py:171
↓ 1 callersFunctiondenoise_controlnet
( model: Flux, controlnet:None, # model input img: Tensor, img_ids: Tensor, txt: Tenso
flux-npu/src/flux/sampling.py:173
↓ 1 callersFunctiondistill_loss
(args, intermediate_double_s, intermediate_double_t, intermediate_single_s, intermediate_single_t)
flux-mini/train_flux_distill.py:65
↓ 1 callersFunctiondistill_loss
(args, intermediate_double_s, intermediate_double_t, intermediate_single_s, intermediate_single_t)
flux-npu/train_flux_distill.py:67
↓ 1 callersMethodforward
( self, prompt, width, height, guidance, num_steps, se
flux-mini/src/flux/xflux_pipeline.py:256
↓ 1 callersMethodforward
( self, prompt, width, height, guidance, num_steps, se
flux-npu/src/flux/xflux_pipeline.py:255
↓ 1 callersFunctionget_lin_function
( x1: float = 256, y1: float = 0.5, x2: float = 4096, y2: float = 1.15 )
flux-mini/src/flux/sampling.py:70
↓ 1 callersFunctionget_lin_function
( x1: float = 256, y1: float = 0.5, x2: float = 4096, y2: float = 1.15 )
flux-npu/src/flux/sampling.py:70
↓ 1 callersFunctionget_lora_rank
(checkpoint)
flux-mini/src/flux/util.py:34
↓ 1 callersFunctionget_lora_rank
(checkpoint)
flux-npu/src/flux/util.py:34
↓ 1 callersFunctionget_models
(name: str, device, offload: bool, is_schnell: bool, args)
flux-mini/train_flux_lora.py:49
↓ 1 callersFunctionget_models
(name: str, device, offload: bool, is_schnell: bool, args)
flux-mini/train_flux_distill.py:55
↓ 1 callersFunctionget_models
(name: str, device, offload: bool, is_schnell: bool, args)
flux-npu/train_flux_lora.py:51
↓ 1 callersFunctionget_models
(name: str, device, offload: bool, is_schnell: bool, args)
flux-npu/train_flux_distill.py:57
↓ 1 callersFunctionimage_resize
(img, max_size=512)
flux-mini/image_datasets/dataset.py:10
↓ 1 callersFunctionimage_resize
(img, max_size=512)
flux-npu/image_datasets/dataset.py:10
↓ 1 callersFunctioninference_prompts
(prompts, dit, t5, clip, vae, is_schnell, height=512, width=512, device='cuda:0', dtype=torch.bfloat16
flux-mini/train_flux_lora.py:71
↓ 1 callersFunctioninference_prompts
(prompts, dit_model, t5, clip, vae, is_schnell, height=512, width=512, device='cuda:0', dtype=torch.bf
flux-mini/train_flux_distill.py:113
↓ 1 callersFunctioninference_prompts
(prompts, dit, t5, clip, vae, is_schnell, height=512, width=512, device='npu:0', dtype=torch.float, se
flux-npu/train_flux_lora.py:73
↓ 1 callersFunctioninference_prompts
(prompts, dit_model, t5, clip, vae, is_schnell, height=512, width=512, device='npu:0', dtype=torch.flo
flux-npu/train_flux_distill.py:115
next →1–100 of 314, ranked by callers