Functions — github.com/black-forest-labs/flux

hub / github.com/black-forest-labs/flux / functions

Functions217 in github.com/black-forest-labs/flux

⨍ Functions 217◇ Types & classes 54

↓ 1 callersFunctionprompt_for_hf_auth

()

src/flux/util.py:45

↓ 1 callersFunctionreplace_linear_with_lora

( module: nn.Module, max_rank: int, scale: float = 1.0, )

src/flux/modules/lora.py:5

↓ 1 callersFunctionresize

(img: Image.Image, min_mp: float = 0.5, max_mp: float = 2.0)

demo_st_fill.py:92

↓ 1 callersFunctionrope

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

src/flux/math.py:15

↓ 1 callersMethodtest_txt

(self, txt: str)

src/flux/content_filters.py:144

↓ 1 callersFunctiontime_shift

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

src/flux/sampling.py:277

Method__call__

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

src/flux/util.py:740

Method__call__

( self, prompt: list[str], )

src/flux/trt/engine/clip_engine.py:33

Method__call__

(self, *args, **Kwargs)

src/flux/trt/engine/base_engine.py:157

Method__call__

( self, z: torch.Tensor, )

src/flux/trt/engine/vae_engine.py:28

Method__call__

( self, x: torch.Tensor, )

src/flux/trt/engine/vae_engine.py:44

Method__call__

( self, **kwargs, )

src/flux/trt/engine/transformer_engine.py:58

Method__call__

( self, prompt: list[str], )

src/flux/trt/engine/t5_engine.py:33

Method__call__

(self, img: torch.Tensor)

src/flux/modules/image_embedders.py:21

Method__call__

(self, img: torch.Tensor)

src/flux/modules/image_embedders.py:47

Method__call__

(self, x: Image.Image)

src/flux/modules/image_embedders.py:92

Method__init__

(self, model_name: str, device: str, offload: bool, track_usage: bool)

( self, device: torch.device = torch.device("cpu"), nsfw_threshold: float = 0.85,

src/flux/content_filters.py:60

Method__init__

( self, lora_rank: int = 128, lora_scale: float = 1.0, *args, **kwargs

src/flux/model.py:123

Method__init__

( self, trt_transformer_precision: str, trt_t5_precision: str, max_batch=2,

src/flux/trt/trt_manager.py:57

Method__init__

(self, trt_config: ClipConfig, stream: torch.cuda.Stream, **kwargs)

src/flux/trt/engine/clip_engine.py:25

Method__init__

(self, size: int, device=torch.device("cuda"))

src/flux/trt/engine/base_engine.py:39

Method__init__

(self, trt_config: VAEDecoderConfig, stream: torch.cuda.Stream, **kwargs)

src/flux/trt/engine/vae_engine.py:24

Method__init__

(self, trt_config: VAEEncoderConfig, stream: torch.cuda.Stream, **kwargs)

src/flux/trt/engine/vae_engine.py:40

Method__init__

(self, trt_config: TransformerConfig, stream: torch.cuda.Stream, **kwargs)

src/flux/trt/engine/transformer_engine.py:46

Method__init__

(self, trt_config: T5Config, stream: torch.cuda.Stream, **kwargs)

src/flux/trt/engine/t5_engine.py:25

Method__init__

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

src/flux/modules/layers.py:12

Method__init__

(self, in_dim: int, hidden_dim: int)

src/flux/modules/layers.py:53

Method__init__

(self, dim: int)

src/flux/modules/layers.py:64

Method__init__

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

src/flux/modules/layers.py:88

Method__init__

(self, dim: int, double: bool)

src/flux/modules/layers.py:114

Method__init__

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

src/flux/modules/layers.py:130

Method__init__

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

src/flux/modules/layers.py:200

Method__init__

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

src/flux/modules/layers.py:243

Method__init__

( self, device, min_t: int = 50, max_t: int = 200, )

src/flux/modules/image_embedders.py:37

Method__init__

( self, device, redux_path: str, redux_dim: int = 1152, txt_in_feature

src/flux/modules/image_embedders.py:67

Method__init__

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

src/flux/modules/conditioner.py:6

Method__init__

(self, in_channels: int)

src/flux/modules/autoencoder.py:26

Method__init__

(self, in_channels: int, out_channels: int)

src/flux/modules/autoencoder.py:56

Method__init__

(self, in_channels: int)

src/flux/modules/autoencoder.py:86

Method__init__

(self, in_channels: int)

src/flux/modules/autoencoder.py:99

Method__init__

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

src/flux/modules/autoencoder.py:184

Method__init__

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

src/flux/modules/autoencoder.py:268

Method__init__

(self, params: AutoEncoderParams, sample_z: bool = False)

src/flux/modules/autoencoder.py:283

Method__init__

( self, in_features: int, out_features: int, bias: bool, rank: int,

src/flux/modules/lora.py:35

Method__new__

(cls, *args, **kwargs)

src/flux/trt/engine/base_engine.py:33

Method__post_init__

(self)

src/flux/trt/trt_config/vae_trt_config.py:41

Method__post_init__

(self)

src/flux/trt/trt_config/base_trt_config.py:218

Method__post_init__

(self)

src/flux/trt/trt_config/transformer_trt_config.py:118

Method__str__

(self)

src/flux/trt/engine/base_engine.py:75

Method__str__

(self)

src/flux/trt/engine/base_engine.py:297

Method_capture_cuda_graph

(self)

src/flux/trt/engine/base_engine.py:270

Method_get_onnx_path

(self)

src/flux/trt/trt_config/transformer_trt_config.py:87

Method_get_onnx_path

(self)

src/flux/trt/trt_config/t5_trt_config.py:74

Methodallocate_buffers

( self, shape_dict: dict[str, tuple], device: str | torch.device = "cuda", )

src/flux/trt/engine/base_engine.py:198

Methodcheck_dims

helper function that check the dimentions associated to each input of a TRT engine

src/flux/trt/trt_config/base_trt_config.py:209

Methodcpu

(self)

src/flux/trt/engine/base_engine.py:118

Methodcpu

(self)

src/flux/trt/engine/vae_engine.py:71

Methodcuda

(self)

src/flux/trt/engine/base_engine.py:122

Methodcuda

(self)

src/flux/trt/engine/vae_engine.py:77

Methoddd_to_flux

(self)

src/flux/trt/engine/transformer_engine.py:50

Methoddeallocate

(self)

src/flux/trt/engine/base_engine.py:61

Methoddeallocate_buffers

(self)

src/flux/trt/engine/base_engine.py:249

Functiondecorator

(cls)

src/flux/trt/trt_config/base_trt_config.py:258

Functiondecrement_counter

()

demo_st.py:144

Methoddevice_memory_size

(self)

src/flux/trt/engine/base_engine.py:258

Methoddevice_memory_size

(self)

src/flux/trt/engine/vae_engine.py:90

Methodflux_to_dd

(self)

src/flux/trt/engine/transformer_engine.py:54

Methodforward

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

src/flux/model.py:84

Methodforward

(self, ids: Tensor)

src/flux/modules/layers.py:18

Methodforward

(self, x: Tensor)

src/flux/modules/layers.py:59

Methodforward

(self, x: Tensor)

src/flux/modules/layers.py:68

Methodforward

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

src/flux/modules/layers.py:81

Methodforward

(self, x: Tensor, pe: Tensor)

src/flux/modules/layers.py:97

Methodforward

(self, vec: Tensor)

src/flux/modules/layers.py:120

Methodforward

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

src/flux/modules/layers.py:158

Methodforward

(self, x: Tensor, vec: Tensor, pe: Tensor)

src/flux/modules/layers.py:227

Methodforward

(self, x: Tensor, vec: Tensor)

src/flux/modules/layers.py:249

Methodforward

(self, text: list[str])

src/flux/modules/conditioner.py:21

Methodforward

(self, x: Tensor)

src/flux/modules/autoencoder.py:51

Methodforward

(self, x)

src/flux/modules/autoencoder.py:69

Methodforward

(self, x: Tensor)

src/flux/modules/autoencoder.py:91

Methodforward

(self, x: Tensor)

src/flux/modules/autoencoder.py:103

Methodforward

(self, x: Tensor)

src/flux/modules/autoencoder.py:159

Methodforward

(self, z: Tensor)

src/flux/modules/autoencoder.py:237

Methodforward

(self, z: Tensor)

src/flux/modules/autoencoder.py:273

Methodforward

(self, x: Tensor)

src/flux/modules/autoencoder.py:317

Methodforward

(self, input: torch.Tensor)

src/flux/modules/lora.py:88

Methodfrom_args

( cls, model_name: str, **kwargs, )

src/flux/trt/trt_config/vae_trt_config.py:79

Methodfrom_args

(cls, model_name: str, **kwargs)

src/flux/trt/trt_config/vae_trt_config.py:183

Methodfrom_args

( cls, model_name: str, **kwargs, )

src/flux/trt/trt_config/clip_trt_config.py:35

Methodfrom_args

(cls, model_name: str, *args, **kwargs)

src/flux/trt/trt_config/base_trt_config.py:174

Methodfrom_args

( cls, model_name, **kwargs, )

src/flux/trt/trt_config/transformer_trt_config.py:57

Methodgenerate_image

( self, width, height, num_steps, guidance, seed, prom

demo_gr.py:51

Methodget_input_profile

( self, batch_size: int, image_height: int | None, image_width: int | None,

src/flux/trt/trt_config/vae_trt_config.py:128

Methodget_input_profile

( self, batch_size: int, image_height: int | None, image_width: int | None,

src/flux/trt/trt_config/vae_trt_config.py:229

Methodget_input_profile

( self, batch_size: int, image_height=None, image_width=None, )

src/flux/trt/trt_config/clip_trt_config.py:51

Methodget_input_profile

Generate max and min shape that each input of a TRT engine can have. Subclasses must implement this method to return a dictionary th

src/flux/trt/trt_config/base_trt_config.py:178

Methodget_input_profile

( self, batch_size: int, image_height: int | None, image_width: int | None,

src/flux/trt/trt_config/transformer_trt_config.py:219

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