MCPcopy Create free account

hub / github.com/YesianRohn/TextSSR / functions

Functions13,352 in github.com/YesianRohn/TextSSR

⨍ Functions 13,352◇ Types & classes 1,980↳ Endpoints 6
↓ 7 callersMethodload_lora_into_unet
This will load the LoRA layers specified in `state_dict` into `unet`. Parameters: state_dict (`dict`): A
diffusers/src/diffusers/loaders/lora_pipeline.py:265
↓ 7 callersMethodpost_process_latents
(self, prior_latents)
diffusers/src/diffusers/models/transformers/prior_transformer.py:378
↓ 7 callersFunctionrenew_attention_paths
Updates paths inside attentions to the new naming scheme (local renaming)
diffusers/scripts/convert_i2vgen_to_diffusers.py:79
↓ 7 callersFunctionrenew_attention_paths
Updates paths inside attentions to the new naming scheme (local renaming)
diffusers/scripts/convert_ms_text_to_video_to_diffusers.py:75
↓ 7 callersFunctionsegment
(a, n)
diffusers/src/diffusers/pipelines/deprecated/spectrogram_diffusion/midi_utils.py:339
↓ 7 callersFunctionshave_segments
Removes segments. Positive values shave the first segments, negative shave the last segments.
diffusers/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py:69
↓ 7 callersMethodsigma_to_t
Gets scaled timesteps from the Karras sigmas for input to the consistency model. Args: sigmas (`float` or `np.ndarray`):
diffusers/src/diffusers/schedulers/scheduling_consistency_models.py:154
↓ 7 callersMethodstep_correct
Correct the predicted sample based on the `model_output` of the network. This is often run repeatedly after making the prediction for
diffusers/src/diffusers/schedulers/scheduling_sde_ve.py:228
↓ 7 callersMethodto
(self, torch_device: Optional[Union[str, torch.device]] = None, silence_dtype_warnings: bool = False)
diffusers/examples/community/stable_diffusion_tensorrt_txt2img.py:836
↓ 7 callersFunctiontokenize_prompt
(tokenizer, prompt)
diffusers/examples/dreambooth/train_dreambooth_lora_sdxl.py:923
↓ 7 callersFunctiontokenize_prompt
(tokenizer, prompt, add_special_tokens=False)
diffusers/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py:1232
↓ 7 callersFunctiontokenize_prompt
(tokenizer, prompt)
diffusers/examples/research_projects/scheduled_huber_loss_training/dreambooth/train_dreambooth_lora_sdxl.py:941
↓ 7 callersMethodtransform
(example)
diffusers/examples/consistency_distillation/train_lcm_distill_sd_wds.py:164
↓ 7 callersMethodunfuse_lora
(self)
diffusers/src/diffusers/loaders/peft.py:295
↓ 6 callersMethod__init__
(self, in_dim, out_dim, hidden_dim, drop_p=0.1, eps=1e-12)
diffusers/src/diffusers/pipelines/blip_diffusion/modeling_blip2.py:340
↓ 6 callersMethod__init__
(self)
diffusers/src/diffusers/pipelines/shap_e/renderer.py:490
↓ 6 callersMethod__init__
(self, in_channels, out_channels, mode, enabled=True)
diffusers/src/diffusers/models/unets/unet_stable_cascade.py:114
↓ 6 callersMethod__init__
(self, hidden_dim, logscale=True)
diffusers/src/diffusers/models/autoencoders/autoencoder_oobleck.py:35
↓ 6 callersMethod__init__
(self, in_channels=1024)
diffusers/scripts/convert_consistency_decoder.py:308
↓ 6 callersFunction_chunked_feed_forward
(ff: nn.Module, hidden_states: torch.Tensor, chunk_dim: int, chunk_size: int)
diffusers/src/diffusers/models/attention.py:31
↓ 6 callersMethod_compare_component_configs
(self, pipe, single_file_pipe)
diffusers/tests/single_file/single_file_testing_utils.py:221
↓ 6 callersFunction_copy_linear
(hf_linear, pt_linear)
diffusers/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py:752
↓ 6 callersFunction_copy_linear
(hf_linear, pt_linear)
diffusers/examples/research_projects/promptdiffusion/convert_original_promptdiffusion_to_diffusers.py:820
↓ 6 callersMethod_test_save_load_local
(self)
diffusers/tests/pipelines/deepfloyd_if/__init__.py:251
↓ 6 callersMethod_upload_folder
Uploads all files in `working_dir` to `repo_id`.
diffusers/src/diffusers/utils/hub_utils.py:522
↓ 6 callersFunctionassign_attention_to_checkpoint
(new_checkpoint, unet_state_dict, old_path, new_path, config)
diffusers/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
diffusers/scripts/convert_unidiffuser_to_diffusers.py:110
↓ 6 callersFunctionbroadcast_to_shape_from_left
(x: jnp.ndarray, shape: Tuple[int])
diffusers/src/diffusers/schedulers/scheduling_utils_flax.py:189
↓ 6 callersFunctioncompute_loss_weighting_for_sd3
Computes loss weighting scheme for SD3 training. Courtesy: This was contributed by Rafie Walker in https://github.com/huggingface/diffusers/pull/
diffusers/src/diffusers/training_utils.py:246
↓ 6 callersFunctionconv_attn_to_linear
(checkpoint)
diffusers/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py:223
↓ 6 callersFunctioncreate_ip_adapter_state_dict
(model)
diffusers/tests/models/unets/test_models_unet_2d_condition.py:89
↓ 6 callersFunctiondeps_list
(*pkgs)
diffusers/setup.py:165
↓ 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
diffusers/src/diffusers/pipelines/kandinsky2_2/pipeline_kandinsky2_2_combined.py:410
↓ 6 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
diffusers/src/diffusers/pipelines/flux/pipeline_flux.py:451
↓ 6 callersFunctionflush
Wipes off memory.
diffusers/benchmarks/utils.py:40
↓ 6 callersMethodget_dummy_components
(self, time_cond_proj_dim=None)
diffusers/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py:425
↓ 6 callersMethodget_dummy_components
(self)
diffusers/tests/pipelines/pixart_sigma/test_pixart.py:59
↓ 6 callersMethodget_dummy_components
(self)
diffusers/tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py:70
↓ 6 callersMethodget_dummy_components
(self)
diffusers/tests/pipelines/pixart_alpha/test_pixart.py:54
↓ 6 callersMethodget_dummy_components
(self, time_cond_proj_dim=None)
diffusers/tests/pipelines/animatediff/test_animatediff_sdxl.py:57
↓ 6 callersMethodget_dummy_inputs
(self, device, seed=0)
diffusers/tests/pipelines/stable_diffusion_3/test_pipeline_stable_diffusion_3_inpaint.py:111
↓ 6 callersMethodget_dummy_inputs
(self, device, seed=0)
diffusers/tests/pipelines/stable_diffusion_3/test_pipeline_stable_diffusion_3_img2img.py:109
↓ 6 callersMethodget_dummy_inputs
(self, device, seed=0, height=64, width=64)
diffusers/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py:431
↓ 6 callersMethodget_dummy_inputs
(self, device, seed=0, height=64, width=64, num_images=1)
diffusers/tests/pipelines/stable_diffusion_adapter/test_stable_diffusion_adapter.py:231
↓ 6 callersMethodget_dummy_inputs
(self, device, seed=0)
diffusers/tests/pipelines/stable_diffusion_2/test_stable_diffusion.py:141
↓ 6 callersMethodget_dummy_inputs
(self, device, seed=0)
diffusers/tests/pipelines/animatediff/test_animatediff_sdxl.py:135
↓ 6 callersMethodget_generator
(self, seed)
diffusers/tests/pipelines/test_pipelines_common.py:900
↓ 6 callersMethodget_inputs
(self, seed=0)
diffusers/tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py:284
↓ 6 callersMethodget_models
(self, decay=0.9999)
diffusers/tests/others/test_ema.py:38
↓ 6 callersMethodget_models
(self, decay=0.9999)
diffusers/tests/others/test_ema.py:171
↓ 6 callersMethodget_scheduler_config
(self, **kwargs)
diffusers/tests/schedulers/test_scheduler_edm_dpmsolver_multistep.py:17
↓ 6 callersMethodget_scheduler_config
(self, **kwargs)
diffusers/tests/schedulers/test_scheduler_euler.py:13
↓ 6 callersMethodinvert
( self, image_path: str, prompt: str, num_inner_steps=10, early_stop_epsilon=1e-6, num_inference_steps
diffusers/examples/community/pipeline_null_text_inversion.py:178
↓ 6 callersFunctionis_k_diffusion_available
()
diffusers/src/diffusers/utils/import_utils.py:395
↓ 6 callersMethodprepare_image
r""" Prepares the input image for processing. Args: image (torch.Tensor or PIL.Image.Image or list): The input image(s).
diffusers/examples/community/stable_diffusion_reference.py:668
↓ 6 callersMethodprepare_init_args_and_inputs_for_common
(self)
diffusers/tests/models/unets/test_unet_2d_blocks.py:155
↓ 6 callersMethodprepare_init_args_and_inputs_for_common
(self)
diffusers/tests/models/unets/test_models_unet_spatiotemporal.py:87
↓ 6 callersMethodpt_to_numpy
Convert a PyTorch tensor to a NumPy image.
diffusers/src/diffusers/image_processor.py:144
↓ 6 callersFunctionrenew_attention_paths
Updates paths inside attentions to the new naming scheme (local renaming)
diffusers/src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py:117
↓ 6 callersFunctionrenew_attention_paths
(old_list, n_shave_prefix_segments=0, in_mid=False)
diffusers/scripts/convert_ddpm_original_checkpoint_to_diffusers.py:35
↓ 6 callersFunctionrenew_attention_paths
Updates paths inside attentions to the new naming scheme (local renaming)
diffusers/scripts/convert_if.py:571
↓ 6 callersFunctionrenew_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
diffusers/scripts/convert_i2vgen_to_diffusers.py:112
↓ 6 callersFunctionrenew_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
diffusers/scripts/convert_ldm_original_checkpoint_to_diffusers.py:35
↓ 6 callersFunctionrenew_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
diffusers/scripts/convert_original_audioldm_to_diffusers.py:56
↓ 6 callersFunctionrenew_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
diffusers/scripts/convert_zero123_to_diffusers.py:208
↓ 6 callersFunctionrenew_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
diffusers/scripts/convert_original_musicldm_to_diffusers.py:57
↓ 6 callersFunctionrenew_resnet_paths
Updates paths inside resnets to the new naming scheme (local renaming)
diffusers/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)
diffusers/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)
diffusers/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
diffusers/src/diffusers/training_utils.py:79
↓ 6 callersFunctionsave_image
(image, name)
diffusers/scripts/convert_consistency_decoder.py:181
↓ 6 callersFunctionset_requires_grad
(model, value)
diffusers/examples/community/clip_guided_images_mixing_stable_diffusion.py:75
↓ 6 callersFunctionset_requires_grad
(model, value)
diffusers/examples/community/clip_guided_stable_diffusion_img2img.py:123
↓ 6 callersFunctionset_requires_grad
(model, value)
diffusers/examples/community/clip_guided_stable_diffusion.py:49
↓ 6 callersFunctionset_resnet_weights
(new_layer, old_checkpoint, index)
diffusers/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.
diffusers/scripts/convert_if.py:539
↓ 6 callersFunctionswap_scale_shift
(weight, dim)
diffusers/src/diffusers/loaders/single_file_utils.py:1667
↓ 6 callersFunctiontext_encoder_attn_modules
(text_encoder)
diffusers/src/diffusers/loaders/__init__.py:33
↓ 6 callersFunctiontext_encoder_mlp_modules
(text_encoder)
diffusers/src/diffusers/models/lora.py:55
↓ 6 callersFunctiontokenize_prompt
(tokenizer, prompt, max_sequence_length)
diffusers/examples/dreambooth/train_dreambooth_lora_flux.py:883
↓ 6 callersFunctiontokenize_prompt
(tokenizer, prompt, max_sequence_length)
diffusers/examples/dreambooth/train_dreambooth_flux.py:846
↓ 6 callersFunctionunwrap_model
(model)
diffusers/examples/dreambooth/train_dreambooth_lora_flux.py:1208
↓ 6 callersFunctionunwrap_model
(model)
diffusers/examples/instruct_pix2pix/train_instruct_pix2pix.py:543
↓ 6 callersFunctionunwrap_model
(model)
diffusers/examples/cogvideo/train_cogvideox_lora.py:1126
↓ 6 callersFunctionupfirdn2d_native
( tensor: torch.Tensor, kernel: torch.Tensor, up: int = 1, down: int = 1, pad: Tuple[int,
diffusers/src/diffusers/models/upsampling.py:415
↓ 6 callersFunctionzero_module
(module: nn.Module)
diffusers/src/diffusers/models/controlnet_sparsectrl.py:785
↓ 5 callersMethod__init__
(self, config: LDMBertConfig)
diffusers/src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py:720
↓ 5 callersMethod__init__
(self, dim: int)
diffusers/src/diffusers/pipelines/unidiffuser/modeling_uvit.py:123
↓ 5 callersMethod__init__
( self, height=224, width=224, patch_size=16, in_channels=3, e
diffusers/src/diffusers/models/transformers/auraflow_transformer_2d.py:50
↓ 5 callersMethod__init__
(self, model, device, max_batch_size, embedding_dim)
diffusers/examples/community/stable_diffusion_tensorrt_inpaint.py:538
↓ 5 callersMethod__init__
(self, model, device, max_batch_size, embedding_dim)
diffusers/examples/community/stable_diffusion_tensorrt_img2img.py:534
↓ 5 callersFunction_extract_into_tensor
(arr, timesteps, broadcast_shape)
diffusers/scripts/convert_consistency_decoder.py:26
↓ 5 callersMethod_fetch_state_dict
( cls, pretrained_model_name_or_path_or_dict, weight_name, use_safetensors,
diffusers/src/diffusers/loaders/lora_base.py:237
↓ 5 callersFunction_get_pipeline_class
( class_obj, config=None, load_connected_pipeline=False, custom_pipeline=None, repo_id=Non
diffusers/src/diffusers/pipelines/pipeline_loading_utils.py:333
↓ 5 callersMethod_get_signature_keys
(cls, obj)
diffusers/src/diffusers/models/modeling_utils.py:967
↓ 5 callersMethod_init_step_index
(self, timestep)
diffusers/src/diffusers/schedulers/scheduling_lcm.py:269
↓ 5 callersMethod_pack_latents
(latents, batch_size, num_channels_latents, height, width)
diffusers/src/diffusers/pipelines/flux/pipeline_flux_inpaint.py:509
↓ 5 callersMethod_pack_latents
(latents, batch_size, num_channels_latents, height, width)
diffusers/examples/community/pipeline_flux_with_cfg.py:481
↓ 5 callersMethod_prepare_latent_image_ids
(batch_size, height, width, device, dtype)
diffusers/examples/community/pipeline_flux_with_cfg.py:467
↓ 5 callersFunction_print_list
Pretty print a list of elements with one line per element and a - starting each line.
diffusers/utils/tests_fetcher.py:828
← previousnext →501–600 of 13,352, ranked by callers