MCPcopy Create free account

hub / github.com/Chen-Yang-Liu/Text2Earth / functions

Functions8,312 in github.com/Chen-Yang-Liu/Text2Earth

↓ 4 callersMethodget_scheduler_config
(self, **kwargs)
tests/schedulers/test_scheduler_unclip.py:12
↓ 4 callersMethodget_scheduler_config
(self, **kwargs)
tests/schedulers/test_scheduler_kdpm2_ancestral.py:13
↓ 4 callersMethodget_scheduler_config
(self, **kwargs)
tests/schedulers/test_scheduler_ddim.py:12
↓ 4 callersMethodget_scheduler_config
(self, **kwargs)
tests/schedulers/test_scheduler_score_sde_ve.py:47
↓ 4 callersMethodget_scheduler_config
(self, **kwargs)
tests/schedulers/test_scheduler_ipndm.py:14
↓ 4 callersMethodget_scheduler_config
(self, **kwargs)
tests/schedulers/test_scheduler_euler_ancestral.py:13
↓ 4 callersFunctionis_clip_model
(checkpoint)
src/diffusers/loaders/single_file_utils.py:355
↓ 4 callersFunctionis_open_clip_model
(checkpoint)
src/diffusers/loaders/single_file_utils.py:369
↓ 4 callersMethodload_module
(name, value)
src/diffusers/loaders/single_file.py:468
↓ 4 callersFunctionload_or_create_model_card
Loads or creates a model card. Args: repo_id_or_path (`str`): The repo id (e.g., "runwayml/stable-diffusion-v1-5") or lo
src/diffusers/utils/hub_utils.py:99
↓ 4 callersMethodmode
(self)
src/diffusers/models/autoencoders/vae.py:819
↓ 4 callersMethodmodel
(sample, t, *args)
tests/schedulers/test_schedulers.py:359
↓ 4 callersFunctionpopulate_model_card
Populates the `model_card` with library name and optional tags.
src/diffusers/utils/hub_utils.py:169
↓ 4 callersMethodprevious_timestep
(self, timestep)
src/diffusers/schedulers/scheduling_lcm.py:645
↓ 4 callersFunctionrearrange_3
(tensor, f)
src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py:64
↓ 4 callersFunctionrearrange_3
(tensor, f)
src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py:38
↓ 4 callersFunctionrearrange_4
(tensor)
src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py:70
↓ 4 callersFunctionrearrange_4
(tensor)
src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py:43
↓ 4 callersFunctionrenew_vae_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py:101
↓ 4 callersMethodreset_device_map
r""" Resets the device maps (if any) to None.
src/diffusers/pipelines/pipeline_utils.py:1168
↓ 4 callersFunctionretrieve_latents
( encoder_output: torch.Tensor, generator: Optional[torch.Generator] = None, sample_mode: str = "sample" )
src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py:108
↓ 4 callersMethodsample
(self, generator: Optional[torch.Generator] = None)
src/diffusers/models/autoencoders/vae.py:780
↓ 4 callersMethodsave_lora_weights
r""" Save the LoRA parameters corresponding to the UNet and text encoder. Arguments: save_directory (`str` or `os.PathLik
src/diffusers/loaders/lora.py:626
↓ 4 callersFunctionset_adapter_layers
(model, enabled=True)
src/diffusers/utils/peft_utils.py:204
↓ 4 callersMethodset_attn_processor
r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o
src/diffusers/models/unets/unet_2d_condition.py:720
↓ 4 callersMethodset_joint_mode
r"""Manually set the generation mode to unconditional joint image-text generation.
src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py:264
↓ 4 callersMethodset_lora_device
Moves the LoRAs listed in `adapter_names` to a target device. Useful for offloading the LoRA to the CPU in case you want to load mult
src/diffusers/loaders/lora.py:1127
↓ 4 callersMethodset_progress_bar_config
(self, **kwargs)
src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_combined.py:198
↓ 4 callersMethodset_progress_bar_config
(self, **kwargs)
src/diffusers/pipelines/kandinsky/pipeline_kandinsky_combined.py:211
↓ 4 callersFunctionset_verbosity
Set the verbosity level for the 🤗 Diffusers' root logger. Args: verbosity (`int`): Logging level which can be one of:
src/diffusers/utils/logging.py:147
↓ 4 callersFunctionupdate_mapping_from_spec
(device_fn_dict: Dict[str, Callable], attribute_name: str)
src/diffusers/utils/testing_utils.py:992
↓ 4 callersFunctionupdate_vae_resnet_ldm_to_diffusers
(keys, new_checkpoint, checkpoint, mapping)
src/diffusers/loaders/single_file_utils.py:730
↓ 3 callersMethod__init__
( self, do_resize: bool = True, vae_scale_factor: int = 8, resample: str = "la
src/diffusers/image_processor.py:85
↓ 3 callersMethod__init__
Initialize an empty AttentionStore :param step_index: used to visualize only a specific step in the diffusion process
src/diffusers/pipelines/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py:122
↓ 3 callersMethod__init__
( self, channels: int, use_conv: bool = False, out_channels: Optional[int] = N
src/diffusers/models/downsampling.py:42
↓ 3 callersMethod__init__
( self, channels: int, use_conv: bool = False, use_conv_transpose: bool = Fals
src/diffusers/models/upsampling.py:41
↓ 3 callersMethod__init__
(self, dim_in: int, dim_out: int, approximate: str = "none", bias: bool = True)
src/diffusers/models/activations.py:76
↓ 3 callersMethod__init__
(self, num_feat, num_grow_ch=32)
Tools/visual_enhancement.py:100
↓ 3 callersMethod_aggregate_and_get_max_attention_per_token
Aggregates the attention for each token and computes the max activation value for each token to alter.
src/diffusers/pipelines/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py:629
↓ 3 callersMethod_alpha_cumprod
(self, t, device)
src/diffusers/schedulers/scheduling_ddpm_wuerstchen.py:118
↓ 3 callersMethod_cast_floating_to
Helper method to cast floating-point values of given parameter `PyTree` to given `dtype`.
src/diffusers/models/modeling_flax_utils.py:71
↓ 3 callersFunction_chunked_feed_forward
(ff: nn.Module, hidden_states: torch.Tensor, chunk_dim: int, chunk_size: int)
src/diffusers/models/attention.py:31
↓ 3 callersMethod_combine_joint
r""" Combines a latent image img_vae of shape (B, C, H, W), a CLIP-embedded image img_clip of shape (B, L_img, clip_img_dim), and a te
src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py:818
↓ 3 callersMethod_compute_loss
Computes the attend-and-excite loss using the maximum attention value for each token.
src/diffusers/pipelines/stable_diffusion_attend_and_excite/pipeline_stable_diffusion_attend_and_excite.py:644
↓ 3 callersFunction_device_agnostic_dispatch
(device: str, dispatch_table: Dict[str, Callable], *args, **kwargs)
src/diffusers/utils/testing_utils.py:950
↓ 3 callersMethod_encode_vae_image
(self, image: torch.Tensor, generator: torch.Generator)
src/diffusers/pipelines/stable_diffusion/pipeline_text2earth_diffusion_inpaint.py:866
↓ 3 callersMethod_encode_vae_image
(self, image: torch.Tensor, generator: torch.Generator)
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py:862
↓ 3 callersMethod_encode_vae_image
(self, image: torch.Tensor, generator: torch.Generator)
src/diffusers/pipelines/paint_by_example/pipeline_paint_by_example.py:354
↓ 3 callersMethod_get_dummy_components
(self)
tests/pipelines/deepfloyd_if/__init__.py:19
↓ 3 callersMethod_get_superresolution_dummy_components
(self)
tests/pipelines/deepfloyd_if/__init__.py:76
↓ 3 callersMethod_get_variance
(self, t, predicted_variance=None, variance_type=None)
src/diffusers/schedulers/scheduling_ddpm.py:326
↓ 3 callersMethod_get_variance
(self, t, predicted_variance=None, variance_type=None)
src/diffusers/schedulers/scheduling_ddpm_parallel.py:337
↓ 3 callersMethod_init_step_index
(self, timestep)
src/diffusers/schedulers/scheduling_lcm.py:269
↓ 3 callersMethod_split
r""" Splits a flattened embedding x of shape (B, C * H * W + clip_img_dim) into two tensors of shape (B, C, H, W) and (B, 1, clip_img_
src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py:774
↓ 3 callersMethod_split_joint
r""" Splits a flattened embedding x of shape (B, C * H * W + clip_img_dim + text_seq_len * text_dim] into (img_vae, img_clip, text) wh
src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py:799
↓ 3 callersFunction_unwrap_model
Unwraps a model.
src/diffusers/pipelines/pipeline_loading_utils.py:230
↓ 3 callersMethodbackward_loop
Perform backward process given list of time steps Args: latents: Latents at time timesteps[0].
src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero_sdxl.py:834
↓ 3 callersMethodbackward_loop
Perform backward process given list of time steps. Args: latents: Latents at time timesteps[0].
src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_zero.py:368
↓ 3 callersMethodbatch_step_no_noise
Batched version of the `step` function, to be able to reverse the SDE for multiple samples/timesteps at once. Also, does not add any
src/diffusers/schedulers/scheduling_ddim_parallel.py:487
↓ 3 callersMethodbinarize
Create a mask. Args: image (`PIL.Image.Image`): The image input, should be a PIL image. Returns
src/diffusers/image_processor.py:408
↓ 3 callersFunctioncheck_if_lora_correctly_set
Checks if the LoRA layers are correctly set with peft
tests/models/unets/test_models_unet_2d_condition.py:79
↓ 3 callersMethodcheck_image
(self, image, prompt, prompt_embeds)
src/diffusers/pipelines/controlnet/pipeline_controlnet.py:758
↓ 3 callersMethodcheck_inputs
( self, prompt, strength, callback_steps, negative_prompt=None,
src/diffusers/pipelines/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py:635
↓ 3 callersMethodcheck_over_forward
(self, time_step=0, **forward_kwargs)
tests/schedulers/test_scheduler_flax.py:440
↓ 3 callersFunctioncheckout_commit
Context manager that checks out a given commit when entered, but gets back to the reference it was at on exit. Args: repo (`git.Repo
utils/tests_fetcher.py:88
↓ 3 callersFunctionclean_doc_toc
Cleans the table of content of the model documentation by removing duplicates and sorting models alphabetically.
utils/check_doc_toc.py:25
↓ 3 callersMethodconvert_model_output
Convert the model output to the corresponding type the data_prediction/noise_prediction algorithm needs. Noise_prediction is designed
src/diffusers/schedulers/scheduling_sasolver.py:398
↓ 3 callersFunctionconvert_open_clip_checkpoint
( text_model, checkpoint, prefix="cond_stage_model.model.", )
src/diffusers/loaders/single_file_utils.py:1224
↓ 3 callersFunctionconvert_state_dict_to_peft
r""" Converts a state dict to the PEFT format The state dict can be from previous diffusers format (`OLD_DIFFUSERS`), or new diffusers format
src/diffusers/utils/state_dict_utils.py:174
↓ 3 callersFunctionconvert_unet_state_dict_to_peft
r""" Converts a state dict from UNet format to diffusers format - i.e. by removing some keys
src/diffusers/utils/state_dict_utils.py:248
↓ 3 callersFunctioncreate_custom_diffusion_layers
(model, mock_weights: bool = True)
tests/models/unets/test_models_unet_2d_condition.py:303
↓ 3 callersFunctiondelete_adapter_layers
(model, adapter_name)
src/diffusers/utils/peft_utils.py:216
↓ 3 callersMethoddelete_adapters
Args: Deletes the LoRA layers of `adapter_name` for the unet and text-encoder(s). adapter_names (`Union[List[str], str]`)
src/diffusers/loaders/lora.py:1048
↓ 3 callersMethoddisable_attention_slicing
r""" Disable sliced attention computation. If `enable_attention_slicing` was previously called, attention is computed in one step.
src/diffusers/pipelines/pipeline_utils.py:1743
↓ 3 callersMethoddisable_free_init
Disables the FreeInit mechanism if enabled.
src/diffusers/pipelines/free_init_utils.py:66
↓ 3 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:155
↓ 3 callersMethoddpm_solver_first_order_update
One step for the first-order DPMSolver (equivalent to DDIM). Args: model_output (`torch.Tensor`): The di
src/diffusers/schedulers/scheduling_dpmsolver_multistep.py:612
↓ 3 callersMethoddummy_model
(self)
tests/schedulers/test_scheduler_flax.py:66
↓ 3 callersMethodenable_forward_chunking
Sets the attention processor to use [feed forward chunking](https://huggingface.co/blog/reformer#2-chunked-feed-forward-layers).
src/diffusers/models/unets/unet_i2vgen_xl.py:381
↓ 3 callersMethodenable_free_init
Enables the FreeInit mechanism as in https://arxiv.org/abs/2312.07537. This implementation has been adapted from the [official repository](ht
src/diffusers/pipelines/free_init_utils.py:27
↓ 3 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:148
↓ 3 callersMethodencode
(self, prefix)
src/diffusers/pipelines/unidiffuser/modeling_text_decoder.py:166
↓ 3 callersMethodencode
Reverse the denoising step process to recover a noisy image from the generated image. Args: images (`List[PIL Image]`):
src/diffusers/pipelines/deprecated/audio_diffusion/pipeline_audio_diffusion.py:271
↓ 3 callersMethodencode_prompt
r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py:205
↓ 3 callersMethodencode_prompt
r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):
src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_stable_diffusion_pix2pix_zero.py:406
↓ 3 callersMethodencode_prompt
r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):
src/diffusers/pipelines/deprecated/stable_diffusion_variants/pipeline_cycle_diffusion.py:284
↓ 3 callersMethodfreeze_base_params
Freeze the weights of the parts belonging to the base UNet2DConditionModel, and leave everything else unfrozen for fine tuning.
src/diffusers/models/controlnet_xs.py:1791
↓ 3 callersMethodfrom_modules
(cls, base_upblock: CrossAttnUpBlock2D, ctrl_upblock: UpBlockControlNetXSAdapter)
src/diffusers/models/controlnet_xs.py:1736
↓ 3 callersMethodfull_loop
(self, **config)
tests/schedulers/test_scheduler_heun.py:44
↓ 3 callersMethodfull_loop
(self, **config)
tests/schedulers/test_scheduler_flax.py:361
↓ 3 callersMethodfull_loop
(self, **config)
tests/schedulers/test_scheduler_flax.py:767
↓ 3 callersMethodfuse_qkv_projections
Enables fused QKV projections. For self-attention modules, all projection matrices (i.e., query, key, value) are fused. For cross-att
src/diffusers/models/unets/unet_i2vgen_xl.py:478
↓ 3 callersMethodgenerate_mask
r""" Generate a latent mask given a mask prompt, a target prompt, and an image. Args: image (`PIL.Image.Image`):
src/diffusers/pipelines/stable_diffusion_diffedit/pipeline_stable_diffusion_diffedit.py:831
↓ 3 callersFunctionget_2d_sincos_pos_embed
grid_size: int of the grid height and width return: pos_embed: [grid_size*grid_size, embed_dim] or [1+grid_size*grid_size, embed_dim] (w/ or
src/diffusers/models/embeddings.py:70
↓ 3 callersMethodget_active_adapters
Gets the list of the current active adapters. Example: ```python from diffusers import DiffusionPipeline p
src/diffusers/loaders/lora.py:1072
↓ 3 callersFunctionget_adapter_name
(model)
src/diffusers/utils/peft_utils.py:195
↓ 3 callersMethodget_dummy_components
(self)
tests/pipelines/controlnet/test_controlnet.py:294
↓ 3 callersMethodget_dummy_components
(self, skip_first_text_encoder=False, time_cond_proj_dim=None)
tests/pipelines/ledits_pp/test_ledits_pp_stable_diffusion_xl.py:58
↓ 3 callersMethodget_dummy_components
(self)
tests/pipelines/ledits_pp/test_ledits_pp_stable_diffusion.py:49
↓ 3 callersMethodget_dummy_components
(self)
tests/pipelines/stable_cascade/test_stable_cascade_prior.py:140
← previousnext →501–600 of 8,312, ranked by callers