Functions — github.com/TencentARC/FluxKits

CodeHub MCPcopy Create free account

hub / github.com/TencentARC/FluxKits / functions

Functions314 in github.com/TencentARC/FluxKits

⨍ Functions 314◇ Types & classes 86

↓ 1 callersFunctionload_controlnet

(name, device, transformer=None)

flux-mini/src/flux/util.py:395

↓ 1 callersFunctionload_controlnet

(name, device, transformer=None)

flux-npu/src/flux/util.py:395

↓ 1 callersFunctionload_flow_model_stu

(name: str, args=None, device: str = "cpu", init_shift = False)

flux-mini/src/flux/util.py:375

↓ 1 callersFunctionload_flow_model_stu

(name: str, args=None, device: str = "npu", init_shift = False)

flux-npu/src/flux/util.py:375

↓ 1 callersFunctionload_from_repo_id

(repo_id, checkpoint_name)

flux-mini/src/flux/util.py:310

↓ 1 callersFunctionload_from_repo_id

(repo_id, checkpoint_name)

flux-npu/src/flux/util.py:310

↓ 1 callersFunctionload_safetensors

flux-mini/src/flux/util.py:26

↓ 1 callersFunctionload_safetensors

flux-npu/src/flux/util.py:26

↓ 1 callersFunctionmain

flux-mini/train_flux_lora.py:93

↓ 1 callersFunctionmain

flux-mini/train_flux_distill.py:143

↓ 1 callersFunctionmain

flux-npu/train_flux_lora.py:95

↓ 1 callersFunctionmain

flux-npu/train_flux_distill.py:145

↓ 1 callersFunctionparse_args

flux-mini/train_flux_lora.py:57

↓ 1 callersFunctionparse_args

flux-mini/train_flux_distill.py:99

↓ 1 callersFunctionparse_args

flux-npu/train_flux_lora.py:59

↓ 1 callersFunctionparse_args

flux-npu/train_flux_distill.py:101

↓ 1 callersFunctionremove_pad

flux-mini/src/flux/util.py:109

↓ 1 callersFunctionremove_pad

flux-npu/src/flux/util.py:109

↓ 1 callersFunctionresize_image_with_pad

(input_image, resolution, skip_hwc3=False, mode='edge')

flux-mini/src/flux/util.py:94

↓ 1 callersFunctionresize_image_with_pad

(input_image, resolution, skip_hwc3=False, mode='edge')

flux-npu/src/flux/util.py:94

↓ 1 callersMethodretrieve

Wait for the generation to finish and retrieve response.

flux-mini/src/flux/api.py:121

↓ 1 callersMethodretrieve

Wait for the generation to finish and retrieve response.

flux-npu/src/flux/api.py:121

↓ 1 callersFunctionrope

(pos, dim: int, theta: int)

flux-mini/src/flux/math.py:13

↓ 1 callersFunctionrope

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

flux-npu/src/flux/math.py:16

↓ 1 callersMethodset_lora

(self, local_path: str = None, repo_id: str = None, name: str = None, lora_weight: int = 0.7)

flux-mini/src/flux/xflux_pipeline.py:97

↓ 1 callersMethodset_lora

(self, local_path: str = None, repo_id: str = None, name: str = None, lora_weight: int = 0.7)

flux-npu/src/flux/xflux_pipeline.py:96

↓ 1 callersMethodset_processor

(self, processor)

flux-mini/src/flux/modules/layers.py:545

↓ 1 callersMethodset_processor

(self, processor)

flux-npu/src/flux/modules/layers.py:545

↓ 1 callersFunctiontime_shift

(mu: float, sigma: float, t: Tensor)

flux-mini/src/flux/sampling.py:66

↓ 1 callersFunctiontime_shift

(mu: float, sigma: float, t: Tensor)

flux-npu/src/flux/sampling.py:66

(self, prompt: str, image_prompt: Image = None, controlnet_

flux-mini/src/flux/xflux_pipeline.py:157

(self, image: Image, width: int, height: int)

flux-mini/src/flux/util.py:131

Adds a predefined watermark to the input image Args: image: ([N,] B, RGB, H, W) in range [-1, 1] Returns:

flux-mini/src/flux/util.py:439

(self, attn, x, pe, **attention_kwargs)

flux-mini/src/flux/modules/layers.py:112

(self, attn, x, pe, **attention_kwargs)

flux-mini/src/flux/modules/layers.py:131

(self, attn, img, txt, vec, pe)

flux-mini/src/flux/modules/layers.py:185

(self, attn, img, txt, vec, pe, image_proj, ip_scale=1.0, **attention_kwargs)

flux-mini/src/flux/modules/layers.py:246

(self, attn, img, txt, vec, pe)

flux-mini/src/flux/modules/layers.py:301

( self, attn: nn.Module, x: Tensor, vec: Tensor, pe: Tensor, i

flux-mini/src/flux/modules/layers.py:413

(self, attn: nn.Module, x: Tensor, vec: Tensor, pe: Tensor)

flux-mini/src/flux/modules/layers.py:467

(self, attn: nn.Module, x: Tensor, vec: Tensor, pe: Tensor)

flux-mini/src/flux/modules/layers.py:489

(self, prompt: str, image_prompt: Image = None, controlnet_

flux-npu/src/flux/xflux_pipeline.py:156

(self, image: Image, width: int, height: int)

flux-npu/src/flux/util.py:131

Adds a predefined watermark to the input image Args: image: ([N,] B, RGB, H, W) in range [-1, 1] Returns:

flux-npu/src/flux/util.py:439

(self, attn, x, pe, **attention_kwargs)

flux-npu/src/flux/modules/layers.py:112

(self, attn, x, pe, **attention_kwargs)

flux-npu/src/flux/modules/layers.py:131

(self, attn, img, txt, vec, pe)

flux-npu/src/flux/modules/layers.py:185

(self, attn, img, txt, vec, pe, image_proj, ip_scale=1.0, **attention_kwargs)

flux-npu/src/flux/modules/layers.py:246

(self, attn, img, txt, vec, pe)

flux-npu/src/flux/modules/layers.py:301

( self, attn: nn.Module, x: Tensor, vec: Tensor, pe: Tensor, i

flux-npu/src/flux/modules/layers.py:413

(self, attn: nn.Module, x: Tensor, vec: Tensor, pe: Tensor)

flux-npu/src/flux/modules/layers.py:467

(self, attn: nn.Module, x: Tensor, vec: Tensor, pe: Tensor)

flux-npu/src/flux/modules/layers.py:489

Method__getitem__

flux-mini/image_datasets/dataset.py:65

Method__getitem__

flux-npu/image_datasets/dataset.py:65

(self, t, s, args)

flux-mini/train_flux_distill.py:137

(self, params: FluxParams, controlnet_depth=2)

flux-mini/src/flux/controlnet.py:39

(self, status_code: int, detail: str | list[dict] | None = None)

flux-mini/src/flux/api.py:13

Manages an image generation request to the API. Args: prompt: Prompt to sample width: Width of the image in

flux-mini/src/flux/api.py:32

(self, model_type, device, offload: bool = False)

flux-mini/src/flux/xflux_pipeline.py:33

(self, name: str, device: str)

flux-mini/src/flux/util.py:115

(self, watermark)

flux-mini/src/flux/util.py:433

(self, params: FluxParams)

flux-mini/src/flux/model.py:41

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

flux-mini/src/flux/modules/layers.py:13

(self, in_dim: int, hidden_dim: int)

flux-mini/src/flux/modules/layers.py:52

(self, dim: int)

flux-mini/src/flux/modules/layers.py:63

(self, in_features, out_features, rank=4, network_alpha=None, device=None, dtype=None)

flux-mini/src/flux/modules/layers.py:86

(self, dim: int, rank=4, network_alpha=None, lora_weight=1)

flux-mini/src/flux/modules/layers.py:124

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

flux-mini/src/flux/modules/layers.py:142

(self, dim: int, double: bool)

flux-mini/src/flux/modules/layers.py:162

(self, dim: int, rank=4, network_alpha=None, lora_weight=1)

flux-mini/src/flux/modules/layers.py:177

(self, context_dim, hidden_dim)

flux-mini/src/flux/modules/layers.py:225

(self, hidden_size: int, num_heads: int, mlp_ratio: float, qkv_bias: bool = False)

flux-mini/src/flux/modules/layers.py:340

(self, context_dim, hidden_dim)

flux-mini/src/flux/modules/layers.py:395

(self, dim: int, rank: int = 4, network_alpha = None, lora_weight: float = 1)

flux-mini/src/flux/modules/layers.py:461

( self, hidden_size: int, num_heads: int, mlp_ratio: float = 4.0, qk_s

flux-mini/src/flux/modules/layers.py:514

(self, hidden_size: int, patch_size: int, out_channels: int)

flux-mini/src/flux/modules/layers.py:566

(self, cross_attention_dim=1024, clip_embeddings_dim=1024, clip_extra_context_tokens=4)

flux-mini/src/flux/modules/layers.py:585

(self, version: str, max_length: int, **hf_kwargs)

flux-mini/src/flux/modules/conditioner.py:7

(self, in_channels: int)

flux-mini/src/flux/modules/autoencoder.py:26

(self, in_channels: int, out_channels: int)

flux-mini/src/flux/modules/autoencoder.py:56

(self, in_channels: int)

flux-mini/src/flux/modules/autoencoder.py:86

(self, in_channels: int)

flux-mini/src/flux/modules/autoencoder.py:99

( self, ch: int, out_ch: int, ch_mult: list, num_res_blocks: int,

flux-mini/src/flux/modules/autoencoder.py:184

(self, sample: bool = True, chunk_dim: int = 1)

flux-mini/src/flux/modules/autoencoder.py:263

(self, params: AutoEncoderParams)

flux-mini/src/flux/modules/autoencoder.py:278

(self, img_dir, img_size=512, caption_type='txt', random_ratio=False)

flux-mini/image_datasets/dataset.py:55

(self, t, s, args)

flux-npu/train_flux_distill.py:139

(self, params: FluxParams, controlnet_depth=2)

flux-npu/src/flux/controlnet.py:38

(self, status_code: int, detail: str | list[dict] | None = None)

flux-npu/src/flux/api.py:13

Manages an image generation request to the API. Args: prompt: Prompt to sample width: Width of the image in

flux-npu/src/flux/api.py:32

(self, model_type, device, offload: bool = False)

flux-npu/src/flux/xflux_pipeline.py:32

(self, name: str, device: str)

flux-npu/src/flux/util.py:115

(self, watermark)

flux-npu/src/flux/util.py:433

(self, params: FluxParams)

flux-npu/src/flux/model.py:41

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

flux-npu/src/flux/modules/layers.py:13

(self, in_dim: int, hidden_dim: int)

flux-npu/src/flux/modules/layers.py:52

(self, dim: int)

flux-npu/src/flux/modules/layers.py:63

(self, in_features, out_features, rank=4, network_alpha=None, device=None, dtype=None)

flux-npu/src/flux/modules/layers.py:86

(self, dim: int, rank=4, network_alpha=None, lora_weight=1)

flux-npu/src/flux/modules/layers.py:124

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

flux-npu/src/flux/modules/layers.py:142

← previous next →101–200 of 314, ranked by callers