Functions — github.com/TencentARC/BrushNet

hub / github.com/TencentARC/BrushNet / functions

Functions9,976 in github.com/TencentARC/BrushNet

⨍ Functions 9,976◇ Types & classes 1,590↳ Endpoints 3

↓ 5 callersFunctionflow_warp

(feature, flow, mask=False, mode="bilinear", padding_mode="zeros")

examples/community/rerender_a_video.py:83

↓ 5 callersMethodfull_loop

(self, scheduler=None, **config)

tests/schedulers/test_scheduler_deis.py:92

↓ 5 callersMethodgelu

(self, gate: torch.Tensor)

src/diffusers/models/activations.py:66

↓ 5 callersMethodget_dummy_components

(self, adapter_type, time_cond_proj_dim=None)

tests/pipelines/stable_diffusion_adapter/test_stable_diffusion_adapter.py:60

↓ 5 callersMethodget_dummy_components

(self)

tests/pipelines/controlnet/test_controlnet_inpaint_sdxl.py:59

↓ 5 callersMethodget_dummy_inputs

(self, device, seed=0, height=64, width=64, num_images=1)

tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py:275

↓ 5 callersMethodget_dummy_inputs

(self, device, seed=0)

tests/pipelines/controlnet/test_controlnet_sdxl.py:785

↓ 5 callersMethodget_dummy_inputs

(self, seed=0)

tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_upscale.py:49

↓ 5 callersMethodget_dummy_inputs

(self, device, seed=0)

tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py:119

↓ 5 callersMethodget_dummy_inputs

(self, device, seed=0)

tests/pipelines/wuerstchen/test_wuerstchen_prior.py:144

↓ 5 callersMethodget_image_processor

(self, repo_id)

tests/pipelines/ip_adapters/test_ip_adapter_stable_diffusion.py:64

↓ 5 callersMethodget_inputs

(self, device="cpu", dtype=torch.float32, seed=0)

tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py:386

↓ 5 callersMethodget_inputs

(self, device, generator_device="cpu", dtype=torch.float32, seed=0)

tests/pipelines/stable_diffusion/test_stable_diffusion.py:1340

↓ 5 callersFunctionget_params_to_save

(params)

examples/textual_inversion/textual_inversion_flax.py:361

↓ 5 callersMethodget_scheduler_config

(self, **kwargs)

tests/schedulers/test_scheduler_edm_euler.py:17

↓ 5 callersMethodget_scheduler_config

(self, **kwargs)

tests/schedulers/test_scheduler_heun.py:13

↓ 5 callersMethodget_scheduler_config

(self, **kwargs)

tests/schedulers/test_scheduler_sasolver.py:15

↓ 5 callersMethodget_scheduler_config

(self, **kwargs)

tests/schedulers/test_scheduler_ddim_parallel.py:26

↓ 5 callersMethodget_scheduler_config

(self, **kwargs)

tests/schedulers/test_scheduler_lms.py:13

↓ 5 callersMethodget_scheduler_config

(self)

tests/schedulers/test_scheduler_flax.py:63

↓ 5 callersMethodget_scheduler_config

(self, **kwargs)

tests/schedulers/test_scheduler_euler.py:13

↓ 5 callersMethodget_sd_image

(self, seed=0, shape=(4, 3, 512, 512), fp16=False)

tests/models/autoencoders/test_models_vae.py:862

↓ 5 callersMethodget_sd_vae_model

(self, model_id="cross-attention/asymmetric-autoencoder-kl-x-1-5", fp16=False)

tests/models/autoencoders/test_models_vae.py:867

↓ 5 callersFunctionget_word_inds

(text: str, word_place: int, tokenizer)

examples/community/pipeline_prompt2prompt.py:1240

↓ 5 callersFunctioninfer_model_type

(original_config, checkpoint=None, model_type=None)

src/diffusers/loaders/single_file_utils.py:310

↓ 5 callersFunctionis_torchsde_available

()

src/diffusers/utils/import_utils.py:379

↓ 5 callersFunctionload_checkpoint_to_model

(checkpoint, model, strict=False)

scripts/convert_kakao_brain_unclip_to_diffusers.py:1033

↓ 5 callersFunctionload_state_dict

Reads a checkpoint file, returning properly formatted errors if they arise.

src/diffusers/models/modeling_utils.py:101

↓ 5 callersMethodmodel

(sample, t, *args)

tests/schedulers/test_schedulers.py:359

↓ 5 callersFunctionmultiply_range

(start_position, multiplier)

examples/community/lpw_stable_diffusion_onnx.py:120

↓ 5 callersFunctionmultiply_range

(start_position, multiplier)

examples/community/lpw_stable_diffusion_xl.py:105

↓ 5 callersFunctionmultiply_range

(start_position, multiplier)

examples/community/lpw_stable_diffusion.py:93

↓ 5 callersFunctiononnx_export

( model, model_args: tuple, output_path: Path, ordered_input_names, output_names, dyna

scripts/convert_stable_diffusion_controlnet_to_onnx.py:187

↓ 5 callersFunctiononnx_export

( model, model_args: tuple, output_path: Path, ordered_input_names, output_names, dyna

scripts/convert_stable_diffusion_checkpoint_to_onnx.py:31

↓ 5 callersMethodprepare_init_args_and_inputs_for_common

(self)

tests/models/unets/test_models_unet_3d_condition.py:60

↓ 5 callersFunctionprior

(*, args, checkpoint_map_location)

scripts/convert_shap_e_to_diffusers.py:930

↓ 5 callersFunctionrearrange_dims

(tensor: torch.Tensor)

src/diffusers/models/resnet.py:409

↓ 5 callersFunctionrenew_resnet_paths

Updates paths inside resnets to the new naming scheme (local renaming)

scripts/convert_original_audioldm2_to_diffusers.py:65

↓ 5 callersFunctionrenew_resnet_paths

Updates paths inside resnets to the new naming scheme (local renaming)

scripts/convert_versatile_diffusion_to_diffusers.py:106

↓ 5 callersFunctionrunEngine

(engine, feed_dict, stream)

examples/community/stable_diffusion_tensorrt_inpaint.py:411

↓ 5 callersFunctionrunEngine

(engine, feed_dict, stream)

examples/community/stable_diffusion_tensorrt_img2img.py:411

↓ 5 callersMethodset_image_to_text_mode

r"""Manually set the generation mode to image-conditioned text generation.

src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py:260

↓ 5 callersMethodset_text_to_image_mode

r"""Manually set the generation mode to text-conditioned image generation.

src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py:256

↓ 5 callersFunctionshave_segments

Removes segments. Positive values shave the first segments, negative shave the last segments.

scripts/convert_original_audioldm2_to_diffusers.py:54

↓ 5 callersFunctionshave_segments

Removes segments. Positive values shave the first segments, negative shave the last segments.

scripts/convert_svd_to_diffusers.py:213

↓ 5 callersFunctionshave_segments

Removes segments. Positive values shave the first segments, negative shave the last segments.

scripts/convert_versatile_diffusion_to_diffusers.py:96

↓ 5 callersFunctionshave_segments

Removes segments. Positive values shave the first segments, negative shave the last segments.

scripts/convert_original_audioldm_to_diffusers.py:45

↓ 5 callersFunctionshave_segments

Removes segments. Positive values shave the first segments, negative shave the last segments.

scripts/convert_zero123_to_diffusers.py:197

↓ 5 callersFunctionshave_segments

Removes segments. Positive values shave the first segments, negative shave the last segments.

scripts/convert_original_musicldm_to_diffusers.py:46

↓ 5 callersFunctionshave_segments

Removes segments. Positive values shave the first segments, negative shave the last segments.

examples/research_projects/promptdiffusion/convert_original_promptdiffusion_to_diffusers.py:71

↓ 5 callersMethodsimulate_backprop

(self, unet)

tests/others/test_ema.py:54

↓ 5 callersFunctionto_sub_blocks

(blocks)

examples/research_projects/controlnetxs/controlnetxs.py:975

↓ 5 callersMethodtransform

(example)

examples/consistency_distillation/train_lcm_distill_sd_wds.py:163

↓ 5 callersFunctionupdate_unet_attention_ldm_to_diffusers

(ldm_keys, new_checkpoint, checkpoint, mapping)

src/diffusers/loaders/single_file_utils.py:561

↓ 5 callersFunctionzero_module

(module)

src/diffusers/models/controlnet.py:865

↓ 4 callersMethod__init__

( self, in_channels: int, out_channels: int, prev_output_channel: int,

src/diffusers/pipelines/audioldm2/modeling_audioldm2.py:1338

↓ 4 callersMethod__init__

(self, c, c_skip=0, kernel_size=3, dropout=0.0)

src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_common.py:31

↓ 4 callersMethod__init__

( self, dim: int, dim_out: Optional[int] = None, mult: int = 4, dropou

src/diffusers/models/attention.py:631

↓ 4 callersMethod__init__

( self, in_features: int, out_features: int, rank: int = 4, network_al

src/diffusers/models/lora.py:196

↓ 4 callersMethod__init__

( self, channels, num_res_blocks: int, hidden_size, hidden_dropout,

src/diffusers/models/unets/uvit_2d.py:308

↓ 4 callersMethod__init__

(self, unet, scheduler)

examples/community/dps_pipeline.py:44

↓ 4 callersMethod_combine

r""" Combines a latent iamge img_vae of shape (B, C, H, W) and a CLIP-embedded image img_clip of shape (B, 1, clip_img_dim) into a sin

src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py:789

↓ 4 callersMethod_encode_prompt

(self, input_ids, attention_mask, text_encoder_use_attention_mask=False)

examples/community/sde_drag.py:348

↓ 4 callersMethod_encode_vae_image

(self, image: torch.Tensor, generator: torch.Generator)

examples/community/pipeline_sdxl_style_aligned.py:1099

↓ 4 callersMethod_get_add_time_ids

( self, original_size, crops_coords_top_left, target_size, dtype, text_encoder_projection_dim=None

examples/community/pipeline_stable_diffusion_xl_ipex.py:633

↓ 4 callersMethod_get_alpha_and_beta

(self, t: torch.Tensor)

examples/community/edict_pipeline.py:85

↓ 4 callersMethod_get_init_keys

(cls)

src/diffusers/configuration_utils.py:452

↓ 4 callersFunctionadd_noise_common

( state: CommonSchedulerState, original_samples: jnp.ndarray, noise: jnp.ndarray, timesteps: jnp.ndarray )

src/diffusers/schedulers/scheduling_utils_flax.py:282

↓ 4 callersFunctionalpha_bar

(time_step)

src/diffusers/schedulers/scheduling_utils_flax.py:214

↓ 4 callersMethodapply_overlay

overlay the inpaint output to the original image

src/diffusers/image_processor.py:612

↓ 4 callersMethodcheck_image

(self, image, prompt, prompt_embeds)

examples/community/pipeline_stable_diffusion_xl_controlnet_adapter.py:495

↓ 4 callersMethodcheck_image

(self, image, prompt, prompt_embeds)

examples/community/pipeline_stable_diffusion_xl_controlnet_adapter_inpaint.py:643

↓ 4 callersMethodcheck_latents_shape

(self, latents_name, latents, expected_shape)

src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py:950

↓ 4 callersMethodcheck_over_configs

(self, time_step=0, **config)

tests/schedulers/test_scheduler_flax.py:646

↓ 4 callersFunctionclamp

(n, smallest, largest)

examples/community/tiled_upscaling.py:52

↓ 4 callersMethodcleanup

(self, return_onnx=False)

examples/community/stable_diffusion_tensorrt_inpaint.py:198

↓ 4 callersMethodcleanup

(self, return_onnx=False)

examples/community/stable_diffusion_tensorrt_txt2img.py:183

↓ 4 callersMethodcleanup

(self, return_onnx=False)

examples/community/stable_diffusion_tensorrt_img2img.py:198

↓ 4 callersFunctioncompare_versions

Args: Compares a library version to some requirement using a given operation. library_or_version (`str` or `packaging.version.Version

src/diffusers/utils/import_utils.py:571

↓ 4 callersFunctioncompute_text_embeddings

(prompt)

examples/dreambooth/train_dreambooth.py:1053

↓ 4 callersFunctioncompute_text_embeddings

(prompt)

examples/dreambooth/train_dreambooth_lora.py:1054

↓ 4 callersFunctionconvert_state_dict

r""" Simply iterates over the state dict and replaces the patterns in `mapping` with the corresponding values. Args: state_dict (`dic

src/diffusers/utils/state_dict_utils.py:133

↓ 4 callersFunctioncosine_distance

(image_embeds, text_embeds)

src/diffusers/pipelines/stable_diffusion_safe/safety_checker.py:25

↓ 4 callersFunctioncosine_distance

(image_embeds, text_embeds)

src/diffusers/pipelines/stable_diffusion/safety_checker.py:26

↓ 4 callersFunctioncreate_dummy_object

Create the code for the dummy object corresponding to `name`.

utils/check_dummies.py:105

↓ 4 callersMethodcreate_state

(self, common: Optional[CommonSchedulerState] = None)

src/diffusers/schedulers/scheduling_ddpm_flax.py:110

↓ 4 callersMethoddisable_slicing

r""" Disable sliced VAE decoding. If `enable_slicing` was previously enabled, this method will go back to computing decoding in one st

src/diffusers/models/autoencoders/autoencoder_kl.py:154

↓ 4 callersMethodenable_sequential_cpu_offload

r""" Offloads all models to CPU using accelerate, significantly reducing memory usage. When called, unet, text_encoder, vae and safety

src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_combined.py:181

↓ 4 callersMethodenable_sequential_cpu_offload

r""" Offloads all models (`unet`, `text_encoder`, `vae`, and `safety checker` state dicts) to CPU using 🤗 Accelerate, significantly re

src/diffusers/pipelines/kandinsky/pipeline_kandinsky_combined.py:195

↓ 4 callersMethodenable_slicing

r""" Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to compute decoding in sev

src/diffusers/models/autoencoders/autoencoder_kl.py:147

↓ 4 callersMethodencode_prompt

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py:261

↓ 4 callersMethodencode_prompt

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py:282

↓ 4 callersMethodevent_type_range

Return [min_id, max_id] for an event type.

src/diffusers/pipelines/deprecated/spectrogram_diffusion/midi_utils.py:200

↓ 4 callersFunctionextract_sub_model

Takes a state dict and returns the state dict for a particular sub-model.

scripts/convert_original_audioldm2_to_diffusers.py:714

↓ 4 callersFunctionfind_backend

Find one (or multiple) backend in a code line of the init.

utils/check_dummies.py:58

↓ 4 callersFunctionfind_backend

Find one (or multiple) backend in a code line of the init.

utils/check_inits.py:50

↓ 4 callersFunctionfreeze_params

(params)

examples/research_projects/intel_opts/textual_inversion_dfq/textual_inversion.py:416

↓ 4 callersMethodfull_loop

(self, **config)

tests/schedulers/test_scheduler_pndm.py:93

↓ 4 callersMethodfull_loop

(self, scheduler=None, **config)

tests/schedulers/test_scheduler_edm_dpmsolver_multistep.py:99

↓ 4 callersMethodfull_loop

(self, **config)

tests/schedulers/test_scheduler_ddim.py:24

← previous next →501–600 of 9,976, ranked by callers