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

↓ 7 callersFunctionupdate_unet_resnet_ldm_to_diffusers

(ldm_keys, new_checkpoint, checkpoint, mapping=None)

src/diffusers/loaders/single_file_utils.py:546

↓ 6 callersMethod__init__

(self, in_dim, out_dim, hidden_dim, drop_p=0.1, eps=1e-12)

src/diffusers/pipelines/blip_diffusion/modeling_blip2.py:340

↓ 6 callersMethod__init__

(self)

src/diffusers/pipelines/shap_e/renderer.py:490

↓ 6 callersMethod__init__

(self, in_channels, out_channels, mode, enabled=True)

src/diffusers/models/unets/unet_stable_cascade.py:114

↓ 6 callersMethod__init__

(self, in_channels=1024)

scripts/convert_consistency_decoder.py:308

↓ 6 callersFunction_add_variant

(weights_name: str, variant: Optional[str] = None)

src/diffusers/utils/hub_utils.py:267

↓ 6 callersFunction_copy_linear

(hf_linear, pt_linear)

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py:752

↓ 6 callersFunction_copy_linear

(hf_linear, pt_linear)

examples/research_projects/promptdiffusion/convert_original_promptdiffusion_to_diffusers.py:820

↓ 6 callersMethod_get_dummy_image_embeds

(self, cross_attention_dim: int = 32)

tests/pipelines/test_pipelines_common.py:231

↓ 6 callersMethod_test_save_load_local

(self)

tests/pipelines/deepfloyd_if/__init__.py:251

↓ 6 callersMethod_upload_folder

Uploads all files in `working_dir` to `repo_id`.

src/diffusers/utils/hub_utils.py:400

↓ 6 callersFunctionassign_attention_to_checkpoint

(new_checkpoint, unet_state_dict, old_path, new_path, config)

scripts/convert_if.py:601

↓ 6 callersFunctionassign_to_checkpoint

This does the final conversion step: take locally converted weights and apply a global renaming to them. It splits attention layers, and take

scripts/convert_unidiffuser_to_diffusers.py:110

↓ 6 callersFunctionbroadcast_to_shape_from_left

(x: jnp.ndarray, shape: Tuple[int])

src/diffusers/schedulers/scheduling_utils_flax.py:191

↓ 6 callersMethodcheck_inputs

( self, prompt, negative_prompt=None, liked=None, disliked=None,

examples/community/pipeline_fabric.py:461

↓ 6 callersFunctionconv_attn_to_linear

(checkpoint)

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py:223

↓ 6 callersFunctiondeps_list

(*pkgs)

setup.py:160

↓ 6 callersMethoddevice

(self)

examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py:794

↓ 6 callersFunctiondevice_view

(t)

examples/community/stable_diffusion_tensorrt_txt2img.py:89

↓ 6 callersMethodenable_freeu

r"""Enables the FreeU mechanism from https://arxiv.org/abs/2309.11497. The suffixes after the scaling factors represent the stage blocks wher

src/diffusers/models/unets/unet_i2vgen_xl.py:444

↓ 6 callersMethodenable_model_cpu_offload

r""" Offloads all models to CPU using accelerate, reducing memory usage with a low impact on performance. Compared to `enable_sequenti

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

↓ 6 callersMethodenable_tiling

r""" Enable tiled VAE decoding. When this option is enabled, the VAE will split the input tensor into tiles to compute decoding and en

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

↓ 6 callersMethodencode_prompt

( self, tokenizers, text_encoders, prompt: str, num_images_per_prompt: int = 1, negative_prompt: str =

tests/pipelines/test_pipelines_common.py:1467

↓ 6 callersFunctionflush

Wipes off memory.

benchmarks/utils.py:40

↓ 6 callersMethodfrom_unet2d

( cls, unet: UNet2DConditionModel, motion_adapter: Optional[MotionAdapter] = None,

src/diffusers/models/unets/unet_motion_model.py:388

↓ 6 callersMethodfuse_projections

(self, fuse=True)

src/diffusers/models/attention_processor.py:701

↓ 6 callersMethodget_dummy_components

(self, time_cond_proj_dim=None)

tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py:421

↓ 6 callersMethodget_dummy_components

(self)

tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py:70

↓ 6 callersMethodget_dummy_components

(self)

tests/pipelines/pixart_alpha/test_pixart.py:54

↓ 6 callersMethodget_dummy_inputs

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

tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py:427

↓ 6 callersMethodget_dummy_inputs

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

tests/pipelines/stable_diffusion_adapter/test_stable_diffusion_adapter.py:231

↓ 6 callersMethodget_dummy_inputs

(self, device, seed=0)

tests/pipelines/stable_diffusion_2/test_stable_diffusion.py:142

↓ 6 callersMethodget_dummy_inputs

(self, device, seed=0)

tests/pipelines/controlnet/test_controlnet.py:379

↓ 6 callersMethodget_generator

(self, seed)

tests/pipelines/test_pipelines_common.py:575

↓ 6 callersMethodget_inputs

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

tests/pipelines/stable_diffusion_2/test_stable_diffusion.py:545

↓ 6 callersMethodget_inputs

(self, seed=0)

tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py:279

↓ 6 callersMethodget_models

(self, decay=0.9999)

tests/others/test_ema.py:38

↓ 6 callersMethodget_scheduler_config

(self, **kwargs)

tests/schedulers/test_scheduler_edm_dpmsolver_multistep.py:17

↓ 6 callersFunctionis_k_diffusion_available

()

src/diffusers/utils/import_utils.py:351

↓ 6 callersFunctionis_torch_xla_available

()

src/diffusers/utils/import_utils.py:303

↓ 6 callersMethodlinear

(x)

examples/community/dps_pipeline.py:203

↓ 6 callersMethodprepare_extra_step_kwargs

(self, generator, eta)

examples/community/instaflow_one_step.py:406

↓ 6 callersMethodprepare_image

r""" Prepares the input image for processing. Args: image (torch.Tensor or PIL.Image.Image or list): The input image(s).

examples/community/stable_diffusion_reference.py:658

↓ 6 callersMethodprepare_init_args_and_inputs_for_common

(self)

tests/models/unets/test_unet_2d_blocks.py:155

↓ 6 callersMethodprepare_init_args_and_inputs_for_common

(self)

tests/models/unets/test_models_unet_spatiotemporal.py:85

↓ 6 callersMethodpt_to_numpy

Convert a PyTorch tensor to a NumPy image.

src/diffusers/image_processor.py:125

↓ 6 callersMethodremove_all_hooks

(self)

src/diffusers/pipelines/deepfloyd_if/pipeline_if.py:159

↓ 6 callersFunctionrenew_attention_paths

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

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py:117

↓ 6 callersFunctionrenew_attention_paths

(old_list, n_shave_prefix_segments=0, in_mid=False)

scripts/convert_ddpm_original_checkpoint_to_diffusers.py:35

↓ 6 callersFunctionrenew_attention_paths

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

scripts/convert_if.py:571

↓ 6 callersFunctionrenew_resnet_paths

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

scripts/convert_i2vgen_to_diffusers.py:112

↓ 6 callersFunctionrenew_resnet_paths

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

scripts/convert_ldm_original_checkpoint_to_diffusers.py:35

↓ 6 callersFunctionrenew_resnet_paths

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

scripts/convert_original_audioldm_to_diffusers.py:56

↓ 6 callersFunctionrenew_resnet_paths

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

scripts/convert_zero123_to_diffusers.py:207

↓ 6 callersFunctionrenew_resnet_paths

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

scripts/convert_original_musicldm_to_diffusers.py:57

↓ 6 callersFunctionrenew_resnet_paths

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

scripts/convert_ms_text_to_video_to_diffusers.py:117

↓ 6 callersFunctionrenew_resnet_paths

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

examples/research_projects/promptdiffusion/convert_original_promptdiffusion_to_diffusers.py:81

↓ 6 callersFunctionrenew_vae_attention_paths

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

src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py:138

↓ 6 callersFunctionresolve_interpolation_mode

Maps a string describing an interpolation function to the corresponding torchvision `InterpolationMode` enum. The full list of supported enum

src/diffusers/training_utils.py:76

↓ 6 callersFunctionsave_image

(image, name)

scripts/convert_consistency_decoder.py:181

↓ 6 callersMethodset_attn_processor

r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o

src/diffusers/pipelines/wuerstchen/modeling_wuerstchen_prior.py:94

↓ 6 callersFunctionset_requires_grad

(model, value)

examples/community/clip_guided_images_mixing_stable_diffusion.py:75

↓ 6 callersFunctionset_requires_grad

(model, value)

examples/community/clip_guided_stable_diffusion_img2img.py:123

↓ 6 callersFunctionset_requires_grad

(model, value)

examples/community/clip_guided_stable_diffusion.py:49

↓ 6 callersFunctionset_resnet_weights

(new_layer, old_checkpoint, index)

scripts/convert_ncsnpp_original_checkpoint_to_diffusers.py:63

↓ 6 callersFunctionshave_segments

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

scripts/convert_if.py:539

↓ 6 callersMethodto

(self, torch_device: Optional[Union[str, torch.device]] = None, silence_dtype_warnings: bool = False)

examples/community/stable_diffusion_tensorrt_inpaint.py:806

↓ 6 callersMethodto

(self, torch_device: Optional[Union[str, torch.device]] = None, silence_dtype_warnings: bool = False)

examples/community/stable_diffusion_tensorrt_img2img.py:806

↓ 6 callersFunctionunwrap_model

(model)

examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py:1379

↓ 6 callersFunctionunwrap_model

(model)

examples/instruct_pix2pix/train_instruct_pix2pix.py:498

↓ 6 callersMethodupcast_vae

(self)

examples/community/pipeline_demofusion_sdxl.py:605

↓ 6 callersFunctionupfirdn2d_native

( tensor: torch.Tensor, kernel: torch.Tensor, up: int = 1, down: int = 1, pad: Tuple[int,

src/diffusers/models/upsampling.py:360

↓ 6 callersFunctionzero_module

(module)

src/diffusers/models/brushnet.py:928

↓ 5 callersMethod__init__

(self, config: LDMBertConfig)

src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py:720

↓ 5 callersMethod__init__

(self, dim: int)

src/diffusers/pipelines/unidiffuser/modeling_uvit.py:123

↓ 5 callersMethod__init__

(self, model, device, max_batch_size, embedding_dim)

examples/community/stable_diffusion_tensorrt_inpaint.py:553

↓ 5 callersMethod__init__

(self, model, device, max_batch_size, embedding_dim)

examples/community/stable_diffusion_tensorrt_img2img.py:553

↓ 5 callersFunction_extract_into_tensor

(arr, timesteps, broadcast_shape)

scripts/convert_consistency_decoder.py:26

↓ 5 callersMethod_get_add_time_ids

(self, original_size, crops_coords_top_left, target_size, dtype)

examples/community/pipeline_demofusion_sdxl.py:496

↓ 5 callersFunction_get_pipeline_class

( class_obj, config=None, load_connected_pipeline=False, custom_pipeline=None, repo_id=Non

src/diffusers/pipelines/pipeline_loading_utils.py:295

↓ 5 callersMethod_make_zero_conv

(self, in_channels, out_channels=None)

examples/research_projects/controlnetxs/controlnetxs.py:750

↓ 5 callersFunction_print_list

Pretty print a list of elements with one line per element and a - starting each line.

utils/tests_fetcher.py:828

↓ 5 callersFunction_tile2latent_indices

Given a tile row and column numbers returns the range of latents affected by that tiles in the overall image Returns a tuple with: - Star

examples/community/mixture_tiling.py:72

↓ 5 callersMethodadd_noise

( self, original_samples: torch.FloatTensor, noise: torch.FloatTensor, timeste

src/diffusers/schedulers/scheduling_ddpm.py:501

↓ 5 callersMethodapply_watermark

(self, images: torch.FloatTensor)

src/diffusers/pipelines/stable_diffusion_xl/watermark.py:24

↓ 5 callersMethodcheck_copy_consistency

(self, comment, class_name, class_code, overwrite_result=None)

tests/others/test_check_copies.py:62

↓ 5 callersMethodcheck_over_configs

(self, time_step=0, **config)

tests/schedulers/test_scheduler_pndm.py:25

↓ 5 callersFunctioncheck_peft_version

r""" Checks if the version of PEFT is compatible. Args: version (`str`): The version of PEFT to check against.

src/diffusers/utils/peft_utils.py:251

↓ 5 callersFunctioncreate_ip_adapter_state_dict

(model)

tests/models/unets/test_models_unet_2d_condition.py:59

↓ 5 callersMethoddisable_lora

(self)

src/diffusers/loaders/lora.py:1048

↓ 5 callersMethoddummy_sample

(self)

tests/schedulers/test_schedulers.py:315

↓ 5 callersMethodenable_adapters

Enable adapters that are attached to the model. The model uses `self.active_adapters()` to retrieve the list of adapters to enable.

src/diffusers/loaders/peft.py:144

↓ 5 callersMethodenable_vae_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/pipelines/pipeline_utils.py:1657

↓ 5 callersMethodencode_event

Encode an event to an index.

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

↓ 5 callersMethodencode_prompt

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

src/diffusers/pipelines/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py:410

↓ 5 callersFunctionextract_into_tensor

(a, t, x_shape)

examples/consistency_distillation/train_lcm_distill_lora_sdxl.py:246

↓ 5 callersFunctionextract_into_tensor

(a, t, x_shape)

examples/consistency_distillation/train_lcm_distill_lora_sd_wds.py:404

↓ 5 callersFunctionextract_into_tensor

(a, t, x_shape)

examples/consistency_distillation/train_lcm_distill_sdxl_wds.py:369

↓ 5 callersFunctionextract_into_tensor

(a, t, x_shape)

examples/consistency_distillation/train_lcm_distill_lora_sdxl_wds.py:391

↓ 5 callersFunctionextract_into_tensor

(a, t, x_shape)

examples/consistency_distillation/train_lcm_distill_sd_wds.py:381

← previous next →401–500 of 9,976, ranked by callers