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

(self, dim: int, double: bool)

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

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

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

(self, context_dim, hidden_dim)

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

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

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

(self, context_dim, hidden_dim)

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

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

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

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

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

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

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

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

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

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

flux-npu/src/flux/modules/conditioner.py:8

(self, in_channels: int)

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

(self, in_channels: int, out_channels: int)

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

(self, in_channels: int)

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

(self, in_channels: int)

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

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

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

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

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

(self, params: AutoEncoderParams)

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

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

flux-npu/image_datasets/dataset.py:55

flux-mini/image_datasets/dataset.py:62

flux-npu/image_datasets/dataset.py:62

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

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

Method_set_gradient_checkpointing

(self, module, value=False)

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

Method_set_gradient_checkpointing

(self, module, value=False)

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

Method_set_gradient_checkpointing

(self, module, value=False)

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

Method_set_gradient_checkpointing

(self, module, value=False)

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

Methodattn_processors

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

Methodattn_processors

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

Methodattn_processors

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

Methodattn_processors

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

Generated image as bytes.

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

Generated image as bytes.

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

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

flux-mini/image_datasets/dataset.py:20

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

flux-npu/image_datasets/dataset.py:20

Methodcreate_custom_forward

(module, return_dict=None)

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

Methodcreate_custom_forward

(module, return_dict=None)

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

Methodcreate_custom_forward

(module, return_dict=None)

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

Methodcreate_custom_forward

(module, return_dict=None)

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

Methodcustom_forward

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

Methodcustom_forward

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

Methodcustom_forward

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

Methodcustom_forward

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

Methodfn_recursive_add_processors

(name: str, module: torch.nn.Module, processors)

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

Methodfn_recursive_add_processors

(name: str, module: torch.nn.Module, processors)

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

Methodfn_recursive_add_processors

(name: str, module: torch.nn.Module, processors)

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

Methodfn_recursive_add_processors

(name: str, module: torch.nn.Module, processors)

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

Methodfn_recursive_attn_processor

(name: str, module: torch.nn.Module, processor)

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

Methodfn_recursive_attn_processor

(name: str, module: torch.nn.Module, processor)

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

Methodfn_recursive_attn_processor

(name: str, module: torch.nn.Module, processor)

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

Methodfn_recursive_attn_processor

(name: str, module: torch.nn.Module, processor)

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

( self, img: Tensor, img_ids: Tensor, controlnet_cond: Tensor, txt: Te

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

( self, img: Tensor, img_ids: Tensor, txt: Tensor, txt_ids: Tensor,

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

(self, ids: Tensor)

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

(self, x: Tensor)

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

(self, x: Tensor)

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

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

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

(self, hidden_states)

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

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

(self, vec: Tensor)

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

( self, img: Tensor, txt: Tensor, vec: Tensor, pe: Tensor, ima

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

( self, x: Tensor, vec: Tensor, pe: Tensor, image_proj: Tensor = None,

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

(self, x: Tensor, vec: Tensor)

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

(self, image_embeds)

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

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

(self, x: Tensor)

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

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

(self, x: Tensor)

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

(self, x: Tensor)

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

(self, x: Tensor)

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

(self, z: Tensor)

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

(self, z: Tensor)

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

(self, x: Tensor)

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

( self, img: Tensor, img_ids: Tensor, controlnet_cond: Tensor, txt: Te

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

( self, img: Tensor, img_ids: Tensor, txt: Tensor, txt_ids: Tensor,

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

(self, ids: Tensor)

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

(self, x: Tensor)

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

(self, x: Tensor)

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

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

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

(self, hidden_states)

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

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

(self, vec: Tensor)

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

( self, img: Tensor, txt: Tensor, vec: Tensor, pe: Tensor, ima

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

( self, x: Tensor, vec: Tensor, pe: Tensor, image_proj: Tensor = None,

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

(self, x: Tensor, vec: Tensor)

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

(self, image_embeds)

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

(self, text: list[str])

flux-npu/src/flux/modules/conditioner.py:23

(self, x: Tensor)

flux-npu/src/flux/modules/autoencoder.py:51

flux-npu/src/flux/modules/autoencoder.py:69

(self, x: Tensor)

flux-npu/src/flux/modules/autoencoder.py:91

(self, x: Tensor)

flux-npu/src/flux/modules/autoencoder.py:103

(self, x: Tensor)

flux-npu/src/flux/modules/autoencoder.py:159

(self, z: Tensor)

flux-npu/src/flux/modules/autoencoder.py:237

(self, z: Tensor)

flux-npu/src/flux/modules/autoencoder.py:268

(self, x: Tensor)

flux-npu/src/flux/modules/autoencoder.py:311

Methodget_processor

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

Methodget_processor

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

Methodget_processor

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

Methodget_processor

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

← previous next →201–300 of 314, ranked by callers