MCPcopy Create free account

hub / github.com/cosmicman-cvpr2024/CosmicMan / functions

Functions6,241 in github.com/cosmicman-cvpr2024/CosmicMan

↓ 20 callersMethod_test_xformers_attention_forwardGenerator_pass
( self, test_max_difference=True, test_mean_pixel_difference=True, expected_max_diff=1e-4 )
diffusers/tests/pipelines/test_pipelines_common.py:696
↓ 20 callersMethodfrom_pretrained
r""" Instantiate a pretrained MultiControlNet model from multiple pre-trained controlnet models. The model is set in evaluation mode
diffusers/src/diffusers/pipelines/controlnet/multicontrolnet.py:116
↓ 20 callersFunctionis_flax_available
()
diffusers/src/diffusers/utils/import_utils.py:317
↓ 20 callersMethodtest_output
(self)
diffusers/tests/models/test_unet_2d_blocks.py:208
↓ 19 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
diffusers/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py:168
↓ 19 callersMethodget_dummy_components
(self)
diffusers/tests/pipelines/test_pipelines_common.py:245
↓ 19 callersMethodget_dummy_inputs
(self, device, seed=0)
diffusers/tests/pipelines/stable_diffusion/test_stable_diffusion.py:154
↓ 19 callersMethodnorm_encoder_hidden_states
(self, encoder_hidden_states)
diffusers/src/diffusers/models/attention_processor.py:424
↓ 19 callersFunctionto_np
(tensor)
diffusers/tests/pipelines/test_pipelines_common.py:23
↓ 18 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
diffusers/src/diffusers/models/autoencoder_kl.py:136
↓ 18 callersFunctionget_activation
(act_fn)
diffusers/src/diffusers/models/activations.py:4
↓ 18 callersMethodload_lora_weights
Load LoRA weights specified in `pretrained_model_name_or_path_or_dict` into self.unet and self.text_encoder. All kwargs are forwarde
diffusers/src/diffusers/loaders.py:807
↓ 17 callersMethoddisable_tiling
r""" Disable tiled VAE decoding. If `enable_tiling` was previously enabled, this method will go back to computing decoding in one step
diffusers/src/diffusers/models/autoencoder_kl.py:144
↓ 17 callersMethoddownload
r""" Download and cache a PyTorch diffusion pipeline from pretrained pipeline weights. Parameters: pretrained_model_name
diffusers/src/diffusers/pipelines/pipeline_utils.py:1145
↓ 17 callersMethodenable_gradient_checkpointing
Activates gradient checkpointing for the current model (may be referred to as *activation checkpointing* or *checkpoint activations*
diffusers/src/diffusers/models/modeling_utils.py:207
↓ 17 callersMethodoptimize
(self, onnx_graph)
diffusers/examples/community/stable_diffusion_tensorrt_inpaint.py:450
↓ 17 callersMethodprepare_init_args_and_inputs_for_common
(self)
diffusers/tests/models/test_models_vq.py:51
↓ 17 callersMethodset_processor
(self, processor: "AttnProcessor")
diffusers/src/diffusers/models/attention_processor.py:306
↓ 17 callersMethodstep_plms
Step function propagating the sample with the linear multi-step method. This has one forward pass with multiple times to approximate
diffusers/src/diffusers/schedulers/scheduling_pndm.py:314
↓ 17 callersMethodstep_prk
Step function propagating the sample with the Runge-Kutta method. RK takes 4 forward passes to approximate the solution to the differ
diffusers/src/diffusers/schedulers/scheduling_pndm.py:259
↓ 16 callersMethod__init__
(self, in_channels, out_channels, mid_channels=None)
diffusers/src/diffusers/models/unet_1d_blocks.py:551
↓ 16 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
diffusers/scripts/convert_if.py:656
↓ 16 callersMethodbatch_to_head_dim
(self, tensor)
diffusers/src/diffusers/models/attention_processor.py:331
↓ 16 callersMethodget_dummy_components
(self)
diffusers/tests/pipelines/stable_diffusion/test_stable_diffusion.py:100
↓ 16 callersMethodget_dummy_input
( self, include_temb=True, include_res_hidden_states_tuple=False, include_enco
diffusers/tests/models/test_unet_blocks_common.py:41
↓ 16 callersMethodget_inputs
(self, device, generator_device="cpu", dtype=torch.float32, seed=0)
diffusers/tests/pipelines/stable_diffusion/test_stable_diffusion.py:649
↓ 16 callersFunctionis_onnx_available
()
diffusers/src/diffusers/utils/import_utils.py:333
↓ 15 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
diffusers/scripts/convert_ms_text_to_video_to_diffusers.py:24
↓ 14 callersFunctionassign_to_checkpoint
( paths, checkpoint, old_checkpoint, attention_paths_to_split=None, additional_replacements=None, config=N
diffusers/scripts/convert_ddpm_original_checkpoint_to_diffusers.py:56
↓ 14 callersMethodget_dummy_inputs
(self, device, seed=0)
diffusers/tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py:160
↓ 14 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: Create sinusoidal timestep embeddings. :param timesteps: a 1-D
diffusers/src/diffusers/models/embeddings.py:24
↓ 14 callersFunctionis_safetensors_compatible
Checking for safetensors compatibility: - By default, all models are saved with the default pytorch serialization, so we use the list of defa
diffusers/src/diffusers/pipelines/pipeline_utils.py:138
↓ 14 callersMethodload_attn_procs
r""" Load pretrained attention processor layers into [`UNet2DConditionModel`]. Attention processor layers have to be defined in
diffusers/src/diffusers/loaders.py:177
↓ 14 callersMethodsave_attn_procs
r""" Save an attention processor to a directory so that it can be reloaded using the [`~loaders.UNet2DConditionLoadersMixin.load_attn_
diffusers/src/diffusers/loaders.py:390
↓ 14 callersMethodwalk
Walks through a series of prompts and seeds, interpolating between them and saving the results to disk. Args: prompts (`
diffusers/examples/community/interpolate_stable_diffusion.py:425
↓ 13 callersMethod__init__
(self, num_channels: int, flip_sin_to_cos: bool, downscale_freq_shift: float)
diffusers/src/diffusers/models/embeddings.py:205
↓ 13 callersFunction_get_library_root_logger
()
diffusers/src/diffusers/utils/logging.py:73
↓ 13 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
diffusers/scripts/convert_versatile_diffusion_to_diffusers.py:195
↓ 13 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
diffusers/scripts/convert_original_audioldm_to_diffusers.py:150
↓ 13 callersMethodcheck_over_forward
(self, time_step=0, **forward_kwargs)
diffusers/tests/schedulers/test_schedulers.py:351
↓ 13 callersMethodload_textual_inversion
r""" Load textual inversion embeddings into the text encoder of [`StableDiffusionPipeline`] (both 🤗 Diffusers and Automatic1111 format
diffusers/src/diffusers/loaders.py:530
↓ 13 callersMethodsort
(a, b)
diffusers/src/diffusers/schedulers/scheduling_dpmsolver_sde.py:45
↓ 13 callersMethodto
(self, device)
diffusers/tests/pipelines/stable_diffusion/test_stable_diffusion_inpaint_legacy.py:152
↓ 12 callersMethod_maybe_convert_prompt
r""" Maybe convert a prompt into a "multi vector"-compatible prompt. If the prompt includes a token that corresponds to a multi-vector
diffusers/src/diffusers/loaders.py:500
↓ 12 callersMethodcheck_dims
(self, batch_size, image_height, image_width)
diffusers/examples/community/stable_diffusion_tensorrt_inpaint.py:272
↓ 12 callersMethodcheck_dims
(self, batch_size, image_height, image_width)
diffusers/examples/community/stable_diffusion_tensorrt_img2img.py:271
↓ 12 callersMethodcheck_over_configs
(self, time_step=0, **config)
diffusers/tests/schedulers/test_scheduler_dpm_multi_inverse.py:34
↓ 12 callersMethodget_attention_scores
(self, query, key, attention_mask=None)
diffusers/src/diffusers/models/attention_processor.py:349
↓ 12 callersMethodget_dummy_inputs_with_latents
(self, device, seed=0)
diffusers/tests/pipelines/unidiffuser/test_unidiffuser.py:127
↓ 12 callersMethodget_inputs
(self, device, generator_device="cpu", dtype=torch.float32, seed=0)
diffusers/tests/pipelines/stable_diffusion/test_stable_diffusion_inpaint.py:342
↓ 12 callersFunctionis_bs4_available
()
diffusers/src/diffusers/utils/import_utils.py:385
↓ 12 callersFunctionis_compiled_module
Check whether the module was compiled with torch.compile()
diffusers/src/diffusers/utils/torch_utils.py:80
↓ 12 callersFunctionis_ftfy_available
()
diffusers/src/diffusers/utils/import_utils.py:381
↓ 12 callersMethodnormalize
Normalize an image array to [-1,1].
diffusers/src/diffusers/image_processor.py:106
↓ 12 callersMethodprepare_init_args_and_inputs_for_common
(self)
diffusers/tests/models/test_models_unet_3d_condition.py:101
↓ 12 callersFunctionrenew_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
diffusers/scripts/convert_if.py:556
↓ 11 callersMethod__init__
(self, channels, use_conv=False, use_conv_transpose=False, out_channels=None, name="conv")
diffusers/src/diffusers/models/resnet.py:42
↓ 11 callersMethodcheck_over_configs
(self, time_step=0, **config)
diffusers/tests/schedulers/test_scheduler_dpm_single.py:38
↓ 11 callersMethodcheck_over_configs
(self, time_step=0, **config)
diffusers/tests/schedulers/test_scheduler_dpm_multi.py:39
↓ 11 callersMethodcopy_to
Copy current averaged parameters into given collection of parameters. Args: parameters: Iterable of `torch.nn.Parameter`;
diffusers/examples/research_projects/intel_opts/textual_inversion_dfq/textual_inversion.py:304
↓ 11 callersMethodget_dummy_components
(self)
diffusers/tests/models/test_lora_layers.py:96
↓ 11 callersMethodget_encoder_hidden_states
(self, seed=0, shape=(4, 77, 768), fp16=False)
diffusers/tests/models/test_models_unet_2d_condition.py:908
↓ 11 callersMethodget_latents
(self, seed=0, shape=(4, 4, 64, 64), fp16=False)
diffusers/tests/models/test_models_unet_2d_condition.py:814
↓ 11 callersMethodget_scheduler_config
(self)
diffusers/tests/schedulers/test_schedulers.py:279
↓ 11 callersMethodget_unet_model
(self, fp16=False, model_id="CompVis/stable-diffusion-v1-4")
diffusers/tests/models/test_models_unet_2d_condition.py:819
↓ 11 callersMethodprepare_inputs
(self, prompt: Union[str, List[str]])
diffusers/src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion.py:168
↓ 10 callersMethodcheck_over_configs
(self, time_step=0, **config)
diffusers/tests/schedulers/test_scheduler_flax.py:384
↓ 10 callersMethodcreate
( cls, common: CommonSchedulerState, final_alpha_cumprod: jnp.ndarray, init_no
diffusers/src/diffusers/schedulers/scheduling_pndm_flax.py:53
↓ 10 callersMethoddecode
( self, h: torch.FloatTensor, force_not_quantize: bool = False, return_dict: bool = True )
diffusers/src/diffusers/models/vq_model.py:130
↓ 10 callersMethodget_dummy_components
(self)
diffusers/tests/pipelines/unidiffuser/test_unidiffuser.py:35
↓ 10 callersMethodget_dummy_inputs
(self, seed=0)
diffusers/tests/pipelines/stable_diffusion/test_onnx_stable_diffusion.py:42
↓ 10 callersMethodget_dummy_inputs_by_type
(self, device, seed=0, input_image_type="pt", output_type="np")
diffusers/tests/pipelines/test_pipelines_common.py:56
↓ 10 callersMethodget_inputs
(self, device, generator_device="cpu", dtype=torch.float32, seed=0)
diffusers/tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py:261
↓ 10 callersMethodprepare_latents
( self, image: Image.Image, text_embeds: torch.Tensor, timesteps: torch.Tensor
diffusers/examples/community/edict_pipeline.py:138
↓ 10 callersMethodsave_lora_weights
r""" Save the LoRA parameters corresponding to the UNet and text encoder. Arguments: save_directory (`str` or `os.PathLik
diffusers/src/diffusers/loaders.py:1147
↓ 10 callersFunctionshave_segments
Removes segments. Positive values shave the first segments, negative shave the last segments.
diffusers/scripts/convert_ddpm_original_checkpoint_to_diffusers.py:9
↓ 9 callersMethod__init__
( self, query_dim: int, cross_attention_dim: Optional[int] = None, heads: int
diffusers/src/diffusers/models/attention_processor.py:52
↓ 9 callersFunction_add_variant
(weights_name: str, variant: Optional[str] = None)
diffusers/src/diffusers/utils/hub_utils.py:237
↓ 9 callersFunction_configure_library_root_logger
()
diffusers/src/diffusers/utils/logging.py:77
↓ 9 callersFunction_get_model_file
( pretrained_model_name_or_path, *, weights_name, subfolder, cache_dir, force_download
diffusers/src/diffusers/utils/hub_utils.py:246
↓ 9 callersMethodcheck_dims
(self, batch_size, image_height, image_width)
diffusers/examples/community/stable_diffusion_tensorrt_txt2img.py:257
↓ 9 callersFunctionconvert_resnet
(checkpoint, new_checkpoint, old_prefix, new_prefix, has_skip=False)
diffusers/scripts/convert_consistency_to_diffusers.py:121
↓ 9 callersFunctioncreate_lora_layers
(model, mock_weights: bool = True)
diffusers/tests/models/test_models_unet_2d_condition.py:48
↓ 9 callersMethodget_dummy_components
(self)
diffusers/tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py:78
↓ 9 callersMethodget_dummy_components
(self)
diffusers/tests/pipelines/stable_diffusion/test_stable_diffusion_pix2pix_zero.py:72
↓ 9 callersMethodget_inputs
(self, device, generator_device="cpu", dtype=torch.float32, seed=0)
diffusers/tests/pipelines/stable_diffusion_2/test_stable_diffusion.py:287
↓ 8 callersMethod__init__
( self, in_channels: int, prev_output_channel: int, out_channels: int,
diffusers/src/diffusers/pipelines/versatile_diffusion/modeling_text_unet.py:1470
↓ 8 callersMethod__init__
(self, dim_in: int, dim_out: int, approximate: str = "none")
diffusers/src/diffusers/models/attention.py:458
↓ 8 callersMethod_get_signature_keys
(obj)
diffusers/src/diffusers/pipelines/pipeline_utils.py:1406
↓ 8 callersMethod_test_inference_batch_consistent
( self, batch_sizes=[2, 4, 13], additional_params_copy_to_batched_inputs=["num_inference_steps"] )
diffusers/tests/pipelines/test_pipelines_common.py:365
↓ 8 callersMethodapply_watermark
(self, images: List[PIL.Image.Image], sample_size=None)
diffusers/src/diffusers/pipelines/deepfloyd_if/watermark.py:19
↓ 8 callersMethodfrom_single_file
r""" Instantiate a [`DiffusionPipeline`] from pretrained pipeline weights saved in the `.ckpt` format. The pipeline is set in evaluati
diffusers/src/diffusers/loaders.py:1320
↓ 8 callersMethodget_dummy_inputs
(self, device, seed=0, pil_image=True)
diffusers/tests/pipelines/unclip/test_unclip_image_variation.py:219
↓ 8 callersMethodget_dummy_inputs
(self, device, seed=0)
diffusers/tests/pipelines/stable_diffusion/test_stable_diffusion_inpaint.py:146
↓ 8 callersMethodget_inputs
(self, device, generator_device="cpu", dtype=torch.float32, seed=0)
diffusers/tests/pipelines/audioldm/test_audioldm.py:379
↓ 8 callersMethodget_scheduler_config
(self, **kwargs)
diffusers/tests/schedulers/test_scheduler_unipc.py:19
↓ 8 callersMethodget_scheduler_config
(self, **kwargs)
diffusers/tests/schedulers/test_scheduler_ddpm_parallel.py:25
↓ 8 callersMethodget_scheduler_config
(self, **kwargs)
diffusers/tests/schedulers/test_scheduler_flax.py:350
↓ 8 callersMethodget_scheduler_config
(self, **kwargs)
diffusers/tests/schedulers/test_scheduler_flax.py:635
↓ 8 callersMethodget_sd_image
(self, seed=0, shape=(4, 3, 512, 512), fp16=False)
diffusers/tests/models/test_models_vae.py:187
← previousnext →101–200 of 6,241, ranked by callers