SYMBOL INDEX (4032 symbols across 300 files) FILE: examples/community/bit_diffusion.py function decimal_to_bits (line 15) | def decimal_to_bits(x, bits=BITS): function bits_to_decimal (line 31) | def bits_to_decimal(x, bits=BITS): function ddim_bit_scheduler_step (line 45) | def ddim_bit_scheduler_step( function ddpm_bit_scheduler_step (line 135) | def ddpm_bit_scheduler_step( class BitDiffusion (line 213) | class BitDiffusion(DiffusionPipeline): method __init__ (line 214) | def __init__( method __call__ (line 229) | def __call__( FILE: examples/community/checkpoint_merger.py class CheckpointMergerPipeline (line 20) | class CheckpointMergerPipeline(DiffusionPipeline): method __init__ (line 41) | def __init__(self): method _compare_model_configs (line 45) | def _compare_model_configs(self, dict0, dict1): method _remove_meta_keys (line 56) | def _remove_meta_keys(self, config_dict: Dict): method merge (line 66) | def merge(self, pretrained_model_name_or_path_list: List[Union[str, os... method weighted_sum (line 271) | def weighted_sum(theta0, theta1, theta2, alpha): method sigmoid (line 276) | def sigmoid(theta0, theta1, theta2, alpha): method inv_sigmoid (line 282) | def inv_sigmoid(theta0, theta1, theta2, alpha): method add_difference (line 289) | def add_difference(theta0, theta1, theta2, alpha): FILE: examples/community/clip_guided_stable_diffusion.py class MakeCutouts (line 21) | class MakeCutouts(nn.Module): method __init__ (line 22) | def __init__(self, cut_size, cut_power=1.0): method forward (line 28) | def forward(self, pixel_values, num_cutouts): function spherical_dist_loss (line 42) | def spherical_dist_loss(x, y): function set_requires_grad (line 48) | def set_requires_grad(model, value): class CLIPGuidedStableDiffusion (line 53) | class CLIPGuidedStableDiffusion(DiffusionPipeline): method __init__ (line 59) | def __init__( method enable_attention_slicing (line 91) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 98) | def disable_attention_slicing(self): method freeze_vae (line 101) | def freeze_vae(self): method unfreeze_vae (line 104) | def unfreeze_vae(self): method freeze_unet (line 107) | def freeze_unet(self): method unfreeze_unet (line 110) | def unfreeze_unet(self): method cond_fn (line 114) | def cond_fn( method __call__ (line 183) | def __call__( FILE: examples/community/composable_stable_diffusion.py class ComposableStableDiffusionPipeline (line 43) | class ComposableStableDiffusionPipeline(DiffusionPipeline): method __init__ (line 72) | def __init__( method enable_vae_slicing (line 168) | def enable_vae_slicing(self): method disable_vae_slicing (line 177) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 184) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 207) | def _execution_device(self): method _encode_prompt (line 224) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method run_safety_checker (line 329) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 339) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 347) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 364) | def check_inputs(self, prompt, height, width, callback_steps): method prepare_latents (line 379) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 397) | def __call__( FILE: examples/community/imagic_stable_diffusion.py function preprocess (line 49) | def preprocess(image): class ImagicStableDiffusionPipeline (line 59) | class ImagicStableDiffusionPipeline(DiffusionPipeline): method __init__ (line 87) | def __init__( method enable_attention_slicing (line 108) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 125) | def disable_attention_slicing(self): method train (line 133) | def train( method __call__ (line 334) | def __call__( FILE: examples/community/img2img_inpainting.py function prepare_mask_and_masked_image (line 21) | def prepare_mask_and_masked_image(image, mask): function check_size (line 38) | def check_size(image, height, width): function overlay_inner_image (line 48) | def overlay_inner_image(image, inner_image, paste_offset: Tuple[int] = (... class ImageToImageInpaintingPipeline (line 58) | class ImageToImageInpaintingPipeline(DiffusionPipeline): method __init__ (line 86) | def __init__( method enable_attention_slicing (line 132) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 151) | def disable_attention_slicing(self): method __call__ (line 160) | def __call__( FILE: examples/community/interpolate_stable_diffusion.py function slerp (line 22) | def slerp(t, v0, v1, DOT_THRESHOLD=0.9995): class StableDiffusionWalkPipeline (line 49) | class StableDiffusionWalkPipeline(DiffusionPipeline): method __init__ (line 77) | def __init__( method enable_attention_slicing (line 123) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 142) | def disable_attention_slicing(self): method __call__ (line 151) | def __call__( method embed_text (line 403) | def embed_text(self, text): method get_noise (line 416) | def get_noise(self, seed, dtype=torch.float32, height=512, width=512): method walk (line 425) | def walk( FILE: examples/community/lpw_stable_diffusion.py function parse_prompt_attention (line 61) | def parse_prompt_attention(text): function get_prompts_with_weights (line 147) | def get_prompts_with_weights(pipe: StableDiffusionPipeline, prompt: List... function pad_tokens_and_weights (line 182) | def pad_tokens_and_weights(tokens, weights, max_length, bos, eos, no_bos... function get_unweighted_text_embeddings (line 207) | def get_unweighted_text_embeddings( function get_weighted_text_embeddings (line 247) | def get_weighted_text_embeddings( function preprocess_image (line 377) | def preprocess_image(image): function preprocess_mask (line 387) | def preprocess_mask(mask, scale_factor=8): class StableDiffusionLongPromptWeightingPipeline (line 400) | class StableDiffusionLongPromptWeightingPipeline(StableDiffusionPipeline): method __init__ (line 431) | def __init__( method __init__ (line 456) | def __init__( method __init__additional__ (line 477) | def __init__additional__(self): method _execution_device (line 482) | def _execution_device(self): method _encode_prompt (line 499) | def _encode_prompt( method check_inputs (line 557) | def check_inputs(self, prompt, height, width, strength, callback_steps): method get_timesteps (line 575) | def get_timesteps(self, num_inference_steps, strength, device, is_text... method run_safety_checker (line 588) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 598) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 606) | def prepare_extra_step_kwargs(self, generator, eta): method prepare_latents (line 623) | def prepare_latents(self, image, timestep, batch_size, height, width, ... method __call__ (line 663) | def __call__( method text2img (line 863) | def text2img( method img2img (line 958) | def img2img( method inpaint (line 1052) | def inpaint( FILE: examples/community/lpw_stable_diffusion_onnx.py function parse_prompt_attention (line 78) | def parse_prompt_attention(text): function get_prompts_with_weights (line 164) | def get_prompts_with_weights(pipe, prompt: List[str], max_length: int): function pad_tokens_and_weights (line 199) | def pad_tokens_and_weights(tokens, weights, max_length, bos, eos, no_bos... function get_unweighted_text_embeddings (line 224) | def get_unweighted_text_embeddings( function get_weighted_text_embeddings (line 265) | def get_weighted_text_embeddings( function preprocess_image (line 404) | def preprocess_image(image): function preprocess_mask (line 413) | def preprocess_mask(mask, scale_factor=8): class OnnxStableDiffusionLongPromptWeightingPipeline (line 425) | class OnnxStableDiffusionLongPromptWeightingPipeline(OnnxStableDiffusion... method __init__ (line 435) | def __init__( method __init__ (line 462) | def __init__( method __init__additional__ (line 485) | def __init__additional__(self): method _encode_prompt (line 489) | def _encode_prompt( method check_inputs (line 540) | def check_inputs(self, prompt, height, width, strength, callback_steps): method get_timesteps (line 558) | def get_timesteps(self, num_inference_steps, strength, is_text2img): method run_safety_checker (line 571) | def run_safety_checker(self, image): method decode_latents (line 589) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 600) | def prepare_extra_step_kwargs(self, generator, eta): method prepare_latents (line 617) | def prepare_latents(self, image, timestep, batch_size, height, width, ... method __call__ (line 650) | def __call__( method text2img (line 865) | def text2img( method img2img (line 957) | def img2img( method inpaint (line 1048) | def inpaint( FILE: examples/community/magic_mix.py class MagicMixPipeline (line 19) | class MagicMixPipeline(DiffusionPipeline): method __init__ (line 20) | def __init__( method encode (line 33) | def encode(self, img): method decode (line 40) | def decode(self, latent): method prep_text (line 50) | def prep_text(self, prompt): method __call__ (line 73) | def __call__( FILE: examples/community/multilingual_stable_diffusion.py function detect_language (line 26) | def detect_language(pipe, prompt, batch_size): function translate_prompt (line 41) | def translate_prompt(prompt, translation_tokenizer, translation_model, d... class MultilingualStableDiffusion (line 51) | class MultilingualStableDiffusion(DiffusionPipeline): method __init__ (line 86) | def __init__( method enable_attention_slicing (line 138) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 157) | def disable_attention_slicing(self): method __call__ (line 166) | def __call__( FILE: examples/community/one_step_unet.py class UnetSchedulerOneForwardPipeline (line 7) | class UnetSchedulerOneForwardPipeline(DiffusionPipeline): method __init__ (line 8) | def __init__(self, unet, scheduler): method __call__ (line 13) | def __call__(self): FILE: examples/community/sd_text2img_k_diffusion.py class ModelWrapper (line 30) | class ModelWrapper: method __init__ (line 31) | def __init__(self, model, alphas_cumprod): method apply_model (line 35) | def apply_model(self, *args, **kwargs): class StableDiffusionPipeline (line 44) | class StableDiffusionPipeline(DiffusionPipeline): method __init__ (line 73) | def __init__( method set_sampler (line 113) | def set_sampler(self, scheduler_type: str): method set_scheduler (line 117) | def set_scheduler(self, scheduler_type: str): method enable_attention_slicing (line 122) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 141) | def disable_attention_slicing(self): method enable_sequential_cpu_offload (line 149) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 167) | def _execution_device(self): method _encode_prompt (line 184) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method run_safety_checker (line 289) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 299) | def decode_latents(self, latents): method check_inputs (line 307) | def check_inputs(self, prompt, height, width, callback_steps): method prepare_latents (line 322) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 339) | def __call__( FILE: examples/community/seed_resize_stable_diffusion.py class SeedResizeStableDiffusionPipeline (line 21) | class SeedResizeStableDiffusionPipeline(DiffusionPipeline): method __init__ (line 49) | def __init__( method enable_attention_slicing (line 70) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 89) | def disable_attention_slicing(self): method __call__ (line 98) | def __call__( FILE: examples/community/speech_to_image_diffusion.py class SpeechToImagePipeline (line 29) | class SpeechToImagePipeline(DiffusionPipeline): method __init__ (line 30) | def __init__( method enable_attention_slicing (line 65) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 70) | def disable_attention_slicing(self): method __call__ (line 74) | def __call__( FILE: examples/community/stable_diffusion_comparison.py class StableDiffusionComparisonPipeline (line 25) | class StableDiffusionComparisonPipeline(DiffusionPipeline): method __init__ (line 53) | def __init__( method layers (line 83) | def layers(self) -> Dict[str, Any]: method enable_attention_slicing (line 86) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 103) | def disable_attention_slicing(self): method text2img_sd1_1 (line 112) | def text2img_sd1_1( method text2img_sd1_2 (line 149) | def text2img_sd1_2( method text2img_sd1_3 (line 186) | def text2img_sd1_3( method text2img_sd1_4 (line 223) | def text2img_sd1_4( method _call_ (line 260) | def _call_( FILE: examples/community/stable_diffusion_mega.py class StableDiffusionMegaPipeline (line 26) | class StableDiffusionMegaPipeline(DiffusionPipeline): method __init__ (line 55) | def __init__( method components (line 93) | def components(self) -> Dict[str, Any]: method enable_attention_slicing (line 96) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 115) | def disable_attention_slicing(self): method inpaint (line 124) | def inpaint( method img2img (line 159) | def img2img( method text2img (line 194) | def text2img( FILE: examples/community/stable_unclip.py function _encode_image (line 17) | def _encode_image(self, image, device, num_images_per_prompt, do_classif... class StableUnCLIPPipeline (line 38) | class StableUnCLIPPipeline(DiffusionPipeline): method __init__ (line 39) | def __init__( method _encode_prompt (line 67) | def _encode_prompt( method _execution_device (line 152) | def _execution_device(self): method prepare_latents (line 169) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method to (line 180) | def to(self, torch_device: Optional[Union[str, torch.device]] = None): method __call__ (line 185) | def __call__( FILE: examples/community/text_inpainting.py class TextInpainting (line 25) | class TextInpainting(DiffusionPipeline): method __init__ (line 59) | def __init__( method enable_attention_slicing (line 123) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 142) | def disable_attention_slicing(self): method enable_sequential_cpu_offload (line 150) | def enable_sequential_cpu_offload(self): method _execution_device (line 169) | def _execution_device(self): method __call__ (line 187) | def __call__( FILE: examples/community/tiled_upscaling.py function make_transparency_mask (line 29) | def make_transparency_mask(size, overlap_pixels, remove_borders=[]): function clamp (line 52) | def clamp(n, smallest, largest): function clamp_rect (line 56) | def clamp_rect(rect: [int], min: [int], max: [int]): function add_overlap_rect (line 65) | def add_overlap_rect(rect: [int], overlap: int, image_size: [int]): function squeeze_tile (line 75) | def squeeze_tile(tile, original_image, original_slice, slice_x): function unsqueeze_tile (line 87) | def unsqueeze_tile(tile, original_image_slice): function next_divisible (line 93) | def next_divisible(n, d): class StableDiffusionTiledUpscalePipeline (line 98) | class StableDiffusionTiledUpscalePipeline(StableDiffusionUpscalePipeline): method __init__ (line 125) | def __init__( method _process_tile (line 145) | def _process_tile(self, original_image_slice, x, y, tile_size, tile_bo... method __call__ (line 185) | def __call__( function main (line 282) | def main(): FILE: examples/community/unclip_image_interpolation.py function slerp (line 28) | def slerp(val, low, high): class UnCLIPImageInterpolationPipeline (line 40) | class UnCLIPImageInterpolationPipeline(DiffusionPipeline): method __init__ (line 87) | def __init__( method prepare_latents (line 116) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 128) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method _encode_image (line 192) | def _encode_image(self, image, device, num_images_per_prompt, image_em... method enable_sequential_cpu_offload (line 207) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 233) | def _execution_device(self): method __call__ (line 251) | def __call__( FILE: examples/community/unclip_text_interpolation.py function slerp (line 24) | def slerp(val, low, high): class UnCLIPTextInterpolationPipeline (line 36) | class UnCLIPTextInterpolationPipeline(DiffusionPipeline): method __init__ (line 82) | def __init__( method prepare_latents (line 111) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 123) | def _encode_prompt( method enable_sequential_cpu_offload (line 215) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 242) | def _execution_device(self): method __call__ (line 260) | def __call__( FILE: examples/community/wildcard_stable_diffusion.py function get_filename (line 25) | def get_filename(path: str): function read_wildcard_values (line 30) | def read_wildcard_values(path: str): function grab_wildcard_values (line 35) | def grab_wildcard_values(wildcard_option_dict: Dict[str, List[str]] = {}... function replace_prompt_with_wildcards (line 45) | def replace_prompt_with_wildcards( class WildcardStableDiffusionOutput (line 62) | class WildcardStableDiffusionOutput(StableDiffusionPipelineOutput): class WildcardStableDiffusionPipeline (line 66) | class WildcardStableDiffusionPipeline(DiffusionPipeline): method __init__ (line 111) | def __init__( method __call__ (line 158) | def __call__( FILE: examples/conftest.py function pytest_addoption (line 34) | def pytest_addoption(parser): function pytest_terminal_summary (line 40) | def pytest_terminal_summary(terminalreporter): FILE: examples/dreambooth/train_dreambooth.py function import_model_class_from_model_name_or_path (line 55) | def import_model_class_from_model_name_or_path(pretrained_model_name_or_... function parse_args (line 75) | def parse_args(input_args=None): class DreamBoothDataset (line 368) | class DreamBoothDataset(Dataset): method __init__ (line 374) | def __init__( method __len__ (line 416) | def __len__(self): method __getitem__ (line 419) | def __getitem__(self, index): function collate_fn (line 449) | def collate_fn(examples, with_prior_preservation=False): class PromptDataset (line 471) | class PromptDataset(Dataset): method __init__ (line 474) | def __init__(self, prompt, num_samples): method __len__ (line 478) | def __len__(self): method __getitem__ (line 481) | def __getitem__(self, index): function get_full_repo_name (line 488) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 498) | def main(args): FILE: examples/dreambooth/train_dreambooth_flax.py function parse_args (line 47) | def parse_args(): class DreamBoothDataset (line 221) | class DreamBoothDataset(Dataset): method __init__ (line 227) | def __init__( method __len__ (line 269) | def __len__(self): method __getitem__ (line 272) | def __getitem__(self, index): class PromptDataset (line 300) | class PromptDataset(Dataset): method __init__ (line 303) | def __init__(self, prompt, num_samples): method __len__ (line 307) | def __len__(self): method __getitem__ (line 310) | def __getitem__(self, index): function get_full_repo_name (line 317) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function get_params_to_save (line 327) | def get_params_to_save(params): function main (line 331) | def main(): FILE: examples/dreambooth/train_dreambooth_lora.py function save_model_card (line 62) | def save_model_card(repo_name, images=None, base_model=str, prompt=str, ... function import_model_class_from_model_name_or_path (line 92) | def import_model_class_from_model_name_or_path(pretrained_model_name_or_... function parse_args (line 112) | def parse_args(input_args=None): class DreamBoothDataset (line 407) | class DreamBoothDataset(Dataset): method __init__ (line 413) | def __init__( method __len__ (line 455) | def __len__(self): method __getitem__ (line 458) | def __getitem__(self, index): function collate_fn (line 488) | def collate_fn(examples, with_prior_preservation=False): class PromptDataset (line 510) | class PromptDataset(Dataset): method __init__ (line 513) | def __init__(self, prompt, num_samples): method __len__ (line 517) | def __len__(self): method __getitem__ (line 520) | def __getitem__(self, index): function get_full_repo_name (line 527) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 537) | def main(args): FILE: examples/research_projects/colossalai/train_dreambooth_colossalai.py function import_model_class_from_model_name_or_path (line 34) | def import_model_class_from_model_name_or_path(pretrained_model_name_or_... function parse_args (line 54) | def parse_args(input_args=None): class DreamBoothDataset (line 251) | class DreamBoothDataset(Dataset): method __init__ (line 257) | def __init__( method __len__ (line 299) | def __len__(self): method __getitem__ (line 302) | def __getitem__(self, index): class PromptDataset (line 330) | class PromptDataset(Dataset): method __init__ (line 333) | def __init__(self, prompt, num_samples): method __len__ (line 337) | def __len__(self): method __getitem__ (line 340) | def __getitem__(self, index): function get_full_repo_name (line 347) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function gemini_zero_dpp (line 358) | def gemini_zero_dpp(model: torch.nn.Module, placememt_policy: str = "aut... function main (line 367) | def main(args): FILE: examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py function prepare_mask_and_masked_image (line 41) | def prepare_mask_and_masked_image(image, mask): function random_mask (line 59) | def random_mask(im_shape, ratio=1, mask_full_image=False): function parse_args (line 83) | def parse_args(): class DreamBoothDataset (line 298) | class DreamBoothDataset(Dataset): method __init__ (line 304) | def __init__( method __len__ (line 351) | def __len__(self): method __getitem__ (line 354) | def __getitem__(self, index): class PromptDataset (line 388) | class PromptDataset(Dataset): method __init__ (line 391) | def __init__(self, prompt, num_samples): method __len__ (line 395) | def __len__(self): method __getitem__ (line 398) | def __getitem__(self, index): function get_full_repo_name (line 405) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 415) | def main(): FILE: examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint_lora.py function prepare_mask_and_masked_image (line 37) | def prepare_mask_and_masked_image(image, mask): function random_mask (line 55) | def random_mask(im_shape, ratio=1, mask_full_image=False): function parse_args (line 79) | def parse_args(): class DreamBoothDataset (line 297) | class DreamBoothDataset(Dataset): method __init__ (line 303) | def __init__( method __len__ (line 350) | def __len__(self): method __getitem__ (line 353) | def __getitem__(self, index): class PromptDataset (line 387) | class PromptDataset(Dataset): method __init__ (line 390) | def __init__(self, prompt, num_samples): method __len__ (line 394) | def __len__(self): method __getitem__ (line 397) | def __getitem__(self, index): function get_full_repo_name (line 404) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 414) | def main(): FILE: examples/research_projects/intel_opts/inference_bf16.py function image_grid (line 8) | def image_grid(imgs, rows, cols): FILE: examples/research_projects/intel_opts/textual_inversion/textual_inversion_bf16.py function save_progress (line 60) | def save_progress(text_encoder, placeholder_token_id, accelerator, args,... function parse_args (line 67) | def parse_args(): class TextualInversionDataset (line 273) | class TextualInversionDataset(Dataset): method __init__ (line 274) | def __init__( method __len__ (line 313) | def __len__(self): method __getitem__ (line 316) | def __getitem__(self, i): function get_full_repo_name (line 359) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function freeze_params (line 369) | def freeze_params(params): function main (line 374) | def main(): FILE: examples/research_projects/multi_subject_dreambooth/train_multi_subject_dreambooth.py function import_model_class_from_model_name_or_path (line 39) | def import_model_class_from_model_name_or_path(pretrained_model_name_or_... function parse_args (line 59) | def parse_args(input_args=None): class DreamBoothDataset (line 326) | class DreamBoothDataset(Dataset): method __init__ (line 332) | def __init__( method __len__ (line 387) | def __len__(self): method __getitem__ (line 390) | def __getitem__(self, index): function collate_fn (line 422) | def collate_fn(num_instances, examples, with_prior_preservation=False): class PromptDataset (line 449) | class PromptDataset(Dataset): method __init__ (line 452) | def __init__(self, prompt, num_samples): method __len__ (line 456) | def __len__(self): method __getitem__ (line 459) | def __getitem__(self, index): function get_full_repo_name (line 466) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 476) | def main(args): FILE: examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py function parse_args (line 54) | def parse_args(): function get_full_repo_name (line 316) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 331) | def main(): FILE: examples/research_projects/onnxruntime/textual_inversion/textual_inversion.py function save_progress (line 84) | def save_progress(text_encoder, placeholder_token_id, accelerator, args,... function parse_args (line 91) | def parse_args(): class TextualInversionDataset (line 380) | class TextualInversionDataset(Dataset): method __init__ (line 381) | def __init__( method __len__ (line 420) | def __len__(self): method __getitem__ (line 423) | def __getitem__(self, i): function get_full_repo_name (line 466) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 476) | def main(): FILE: examples/research_projects/onnxruntime/unconditional_image_generation/train_unconditional.py function _extract_into_tensor (line 34) | def _extract_into_tensor(arr, timesteps, broadcast_shape): function parse_args (line 51) | def parse_args(): function get_full_repo_name (line 266) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 276) | def main(args): FILE: examples/test_examples.py class SubprocessCallException (line 37) | class SubprocessCallException(Exception): function run_command (line 41) | def run_command(command: List[str], return_stdout=False): class ExamplesTestsAccelerate (line 62) | class ExamplesTestsAccelerate(unittest.TestCase): method setUpClass (line 64) | def setUpClass(cls): method tearDownClass (line 73) | def tearDownClass(cls): method test_train_unconditional (line 77) | def test_train_unconditional(self): method test_textual_inversion (line 98) | def test_textual_inversion(self): method test_dreambooth (line 122) | def test_dreambooth(self): method test_dreambooth_checkpointing (line 145) | def test_dreambooth_checkpointing(self): method test_text_to_image (line 224) | def test_text_to_image(self): method test_text_to_image_checkpointing (line 248) | def test_text_to_image_checkpointing(self): method test_text_to_image_checkpointing_use_ema (line 328) | def test_text_to_image_checkpointing_use_ema(self): FILE: examples/text_to_image/train_text_to_image.py function parse_args (line 55) | def parse_args(): function get_full_repo_name (line 317) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 332) | def main(): FILE: examples/text_to_image/train_text_to_image_flax.py function parse_args (line 42) | def parse_args(): function get_full_repo_name (line 218) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function get_params_to_save (line 233) | def get_params_to_save(params): function main (line 237) | def main(): FILE: examples/text_to_image/train_text_to_image_lora.py function save_model_card (line 56) | def save_model_card(repo_name, images=None, base_model=str, dataset_name... function parse_args (line 84) | def parse_args(): function get_full_repo_name (line 349) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 364) | def main(): FILE: examples/textual_inversion/textual_inversion.py function log_validation (line 86) | def log_validation(text_encoder, tokenizer, unet, vae, args, accelerator... function save_progress (line 130) | def save_progress(text_encoder, placeholder_token_id, accelerator, args,... function parse_args (line 137) | def parse_args(): class TextualInversionDataset (line 436) | class TextualInversionDataset(Dataset): method __init__ (line 437) | def __init__( method __len__ (line 476) | def __len__(self): method __getitem__ (line 479) | def __getitem__(self, i): function get_full_repo_name (line 522) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 532) | def main(): FILE: examples/textual_inversion/textual_inversion_flax.py function parse_args (line 65) | def parse_args(): class TextualInversionDataset (line 243) | class TextualInversionDataset(Dataset): method __init__ (line 244) | def __init__( method __len__ (line 283) | def __len__(self): method __getitem__ (line 286) | def __getitem__(self, i): function get_full_repo_name (line 329) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function resize_token_embeddings (line 339) | def resize_token_embeddings(model, new_num_tokens, initializer_token_id,... function get_params_to_save (line 359) | def get_params_to_save(params): function main (line 363) | def main(): FILE: examples/unconditional_image_generation/train_unconditional.py function _extract_into_tensor (line 36) | def _extract_into_tensor(arr, timesteps, broadcast_shape): function parse_args (line 54) | def parse_args(): function get_full_repo_name (line 278) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function main (line 288) | def main(args): FILE: scripts/conversion_ldm_uncond.py function convert_ldm_original (line 9) | def convert_ldm_original(checkpoint_path, config_path, output_path): FILE: scripts/convert_dance_diffusion_to_diffusers.py function alpha_sigma_to_t (line 49) | def alpha_sigma_to_t(alpha, sigma): function get_crash_schedule (line 55) | def get_crash_schedule(t): class Object (line 61) | class Object(object): class DiffusionUncond (line 65) | class DiffusionUncond(nn.Module): method __init__ (line 66) | def __init__(self, global_args): function download (line 74) | def download(model_name): function convert_resconv_naming (line 135) | def convert_resconv_naming(name): function convert_attn_naming (line 146) | def convert_attn_naming(name): function rename (line 155) | def rename(input_string, max_depth=13): function rename_orig_weights (line 214) | def rename_orig_weights(state_dict): function transform_conv_attns (line 232) | def transform_conv_attns(new_state_dict, new_k, v): function main (line 252) | def main(args): FILE: scripts/convert_ddpm_original_checkpoint_to_diffusers.py function shave_segments (line 9) | def shave_segments(path, n_shave_prefix_segments=1): function renew_resnet_paths (line 19) | def renew_resnet_paths(old_list, n_shave_prefix_segments=0): function renew_attention_paths (line 35) | def renew_attention_paths(old_list, n_shave_prefix_segments=0, in_mid=Fa... function assign_to_checkpoint (line 56) | def assign_to_checkpoint( function convert_ddpm_checkpoint (line 99) | def convert_ddpm_checkpoint(checkpoint, config): function convert_vq_autoenc_checkpoint (line 235) | def convert_vq_autoenc_checkpoint(checkpoint, config): FILE: scripts/convert_diffusers_to_original_stable_diffusion.py function convert_unet_state_dict (line 92) | def convert_unet_state_dict(unet_state_dict): function reshape_weight_for_sd (line 163) | def reshape_weight_for_sd(w): function convert_vae_state_dict (line 168) | def convert_vae_state_dict(vae_state_dict): function convert_text_enc_state_dict_v20 (line 213) | def convert_text_enc_state_dict_v20(text_enc_dict): function convert_text_enc_state_dict (line 260) | def convert_text_enc_state_dict(text_enc_dict): FILE: scripts/convert_dit_to_diffusers.py function download_model (line 13) | def download_model(model_name): function main (line 26) | def main(args): FILE: scripts/convert_k_upscaler_to_diffusers.py function resnet_to_diffusers_checkpoint (line 13) | def resnet_to_diffusers_checkpoint(resnet, checkpoint, *, diffusers_resn... function self_attn_to_diffusers_checkpoint (line 39) | def self_attn_to_diffusers_checkpoint(checkpoint, *, diffusers_attention... function cross_attn_to_diffusers_checkpoint (line 65) | def cross_attn_to_diffusers_checkpoint( function block_to_diffusers_checkpoint (line 115) | def block_to_diffusers_checkpoint(block, checkpoint, block_idx, block_ty... function unet_to_diffusers_checkpoint (line 175) | def unet_to_diffusers_checkpoint(model, checkpoint): function unet_model_from_original_config (line 217) | def unet_model_from_original_config(original_config): function main (line 269) | def main(args): FILE: scripts/convert_kakao_brain_unclip_to_diffusers.py function prior_model_from_original_config (line 45) | def prior_model_from_original_config(): function prior_original_checkpoint_to_diffusers_checkpoint (line 51) | def prior_original_checkpoint_to_diffusers_checkpoint(model, checkpoint,... function prior_attention_to_diffusers (line 172) | def prior_attention_to_diffusers( function prior_ff_to_diffusers (line 207) | def prior_ff_to_diffusers(checkpoint, *, diffusers_ff_prefix, original_f... function decoder_model_from_original_config (line 255) | def decoder_model_from_original_config(): function decoder_original_checkpoint_to_diffusers_checkpoint (line 261) | def decoder_original_checkpoint_to_diffusers_checkpoint(model, checkpoint): function text_proj_from_original_config (line 330) | def text_proj_from_original_config(): function text_proj_original_checkpoint_to_diffusers_checkpoint (line 343) | def text_proj_original_checkpoint_to_diffusers_checkpoint(checkpoint): function super_res_unet_first_steps_model_from_original_config (line 399) | def super_res_unet_first_steps_model_from_original_config(): function super_res_unet_first_steps_original_checkpoint_to_diffusers_checkpoint (line 405) | def super_res_unet_first_steps_original_checkpoint_to_diffusers_checkpoi... function super_res_unet_last_step_model_from_original_config (line 494) | def super_res_unet_last_step_model_from_original_config(): function super_res_unet_last_step_original_checkpoint_to_diffusers_checkpoint (line 500) | def super_res_unet_last_step_original_checkpoint_to_diffusers_checkpoint... function unet_time_embeddings (line 568) | def unet_time_embeddings(checkpoint, original_unet_prefix): function unet_conv_in (line 584) | def unet_conv_in(checkpoint, original_unet_prefix): function unet_conv_norm_out (line 598) | def unet_conv_norm_out(checkpoint, original_unet_prefix): function unet_conv_out (line 612) | def unet_conv_out(checkpoint, original_unet_prefix): function unet_downblock_to_diffusers_checkpoint (line 626) | def unet_downblock_to_diffusers_checkpoint( function unet_midblock_to_diffusers_checkpoint (line 685) | def unet_midblock_to_diffusers_checkpoint(model, checkpoint, *, original... function unet_upblock_to_diffusers_checkpoint (line 729) | def unet_upblock_to_diffusers_checkpoint( function resnet_to_diffusers_checkpoint (line 799) | def resnet_to_diffusers_checkpoint(checkpoint, *, diffusers_resnet_prefi... function attention_to_diffusers_checkpoint (line 826) | def attention_to_diffusers_checkpoint(checkpoint, *, diffusers_attention... function split_attentions (line 887) | def split_attentions(*, weight, bias, split, chunk_size): function text_encoder (line 919) | def text_encoder(): function prior (line 942) | def prior(*, args, checkpoint_map_location): function decoder (line 966) | def decoder(*, args, checkpoint_map_location): function super_res_unet (line 1001) | def super_res_unet(*, args, checkpoint_map_location): function load_checkpoint_to_model (line 1033) | def load_checkpoint_to_model(checkpoint, model, strict=False): FILE: scripts/convert_ldm_original_checkpoint_to_diffusers.py function shave_segments (line 25) | def shave_segments(path, n_shave_prefix_segments=1): function renew_resnet_paths (line 35) | def renew_resnet_paths(old_list, n_shave_prefix_segments=0): function renew_attention_paths (line 57) | def renew_attention_paths(old_list, n_shave_prefix_segments=0): function assign_to_checkpoint (line 78) | def assign_to_checkpoint( function convert_ldm_checkpoint (line 130) | def convert_ldm_checkpoint(checkpoint, config): FILE: scripts/convert_models_diffuser_to_diffusers.py function unet (line 15) | def unet(hor): function value_function (line 59) | def value_function(): FILE: scripts/convert_ncsnpp_original_checkpoint_to_diffusers.py function convert_ncsnpp_checkpoint (line 25) | def convert_ncsnpp_checkpoint(checkpoint, config): FILE: scripts/convert_stable_diffusion_checkpoint_to_onnx.py function onnx_export (line 31) | def onnx_export( function convert_models (line 71) | def convert_models(model_path: str, output_path: str, opset: int, fp16: ... FILE: scripts/convert_vae_pt_to_diffusers.py function custom_convert_ldm_vae_checkpoint (line 18) | def custom_convert_ldm_vae_checkpoint(checkpoint, config): function vae_pt_to_vae_diffuser (line 119) | def vae_pt_to_vae_diffuser( FILE: scripts/convert_versatile_diffusion_to_diffusers.py function shave_segments (line 96) | def shave_segments(path, n_shave_prefix_segments=1): function renew_resnet_paths (line 106) | def renew_resnet_paths(old_list, n_shave_prefix_segments=0): function renew_vae_resnet_paths (line 128) | def renew_vae_resnet_paths(old_list, n_shave_prefix_segments=0): function renew_attention_paths (line 144) | def renew_attention_paths(old_list, n_shave_prefix_segments=0): function renew_vae_attention_paths (line 165) | def renew_vae_attention_paths(old_list, n_shave_prefix_segments=0): function assign_to_checkpoint (line 195) | def assign_to_checkpoint( function conv_attn_to_linear (line 247) | def conv_attn_to_linear(checkpoint): function create_image_unet_diffusers_config (line 259) | def create_image_unet_diffusers_config(unet_params): function create_text_unet_diffusers_config (line 298) | def create_text_unet_diffusers_config(unet_params): function create_vae_diffusers_config (line 337) | def create_vae_diffusers_config(vae_params): function create_diffusers_scheduler (line 359) | def create_diffusers_scheduler(original_config): function convert_vd_unet_checkpoint (line 369) | def convert_vd_unet_checkpoint(checkpoint, config, unet_key, extract_ema... function convert_vd_vae_checkpoint (line 574) | def convert_vd_vae_checkpoint(checkpoint, config): FILE: scripts/convert_vq_diffusion_to_diffusers.py function vqvae_model_from_original_config (line 61) | def vqvae_model_from_original_config(original_config): function get_down_block_types (line 110) | def get_down_block_types(original_encoder_config): function get_up_block_types (line 131) | def get_up_block_types(original_decoder_config): function coerce_attn_resolutions (line 152) | def coerce_attn_resolutions(attn_resolutions): function coerce_resolution (line 163) | def coerce_resolution(resolution): function vqvae_original_checkpoint_to_diffusers_checkpoint (line 179) | def vqvae_original_checkpoint_to_diffusers_checkpoint(model, checkpoint): function vqvae_encoder_to_diffusers_checkpoint (line 210) | def vqvae_encoder_to_diffusers_checkpoint(model, checkpoint): function vqvae_decoder_to_diffusers_checkpoint (line 310) | def vqvae_decoder_to_diffusers_checkpoint(model, checkpoint): function vqvae_resnet_to_diffusers_checkpoint (line 413) | def vqvae_resnet_to_diffusers_checkpoint(resnet, checkpoint, *, diffuser... function vqvae_attention_to_diffusers_checkpoint (line 440) | def vqvae_attention_to_diffusers_checkpoint(checkpoint, *, diffusers_att... function transformer_model_from_original_config (line 471) | def transformer_model_from_original_config( function transformer_original_checkpoint_to_diffusers_checkpoint (line 537) | def transformer_original_checkpoint_to_diffusers_checkpoint(model, check... function transformer_ada_norm_to_diffusers_checkpoint (line 653) | def transformer_ada_norm_to_diffusers_checkpoint(checkpoint, *, diffuser... function transformer_attention_to_diffusers_checkpoint (line 661) | def transformer_attention_to_diffusers_checkpoint(checkpoint, *, diffuse... function transformer_feedforward_to_diffusers_checkpoint (line 678) | def transformer_feedforward_to_diffusers_checkpoint(checkpoint, *, diffu... function read_config_file (line 690) | def read_config_file(filename): FILE: setup.py function deps_list (line 138) | def deps_list(*pkgs): class DepsTableUpdateCommand (line 142) | class DepsTableUpdateCommand(Command): method initialize_options (line 154) | def initialize_options(self): method finalize_options (line 157) | def finalize_options(self): method run (line 160) | def run(self): FILE: src/diffusers/commands/__init__.py class BaseDiffusersCLICommand (line 19) | class BaseDiffusersCLICommand(ABC): method register_subcommand (line 22) | def register_subcommand(parser: ArgumentParser): method run (line 26) | def run(self): FILE: src/diffusers/commands/diffusers_cli.py function main (line 21) | def main(): FILE: src/diffusers/commands/env.py function info_command_factory (line 25) | def info_command_factory(_): class EnvironmentCommand (line 29) | class EnvironmentCommand(BaseDiffusersCLICommand): method register_subcommand (line 31) | def register_subcommand(parser: ArgumentParser): method run (line 35) | def run(self): method format_dict (line 83) | def format_dict(d): FILE: src/diffusers/configuration_utils.py class FrozenDict (line 42) | class FrozenDict(OrderedDict): method __init__ (line 43) | def __init__(self, *args, **kwargs): method __delitem__ (line 51) | def __delitem__(self, *args, **kwargs): method setdefault (line 54) | def setdefault(self, *args, **kwargs): method pop (line 57) | def pop(self, *args, **kwargs): method update (line 60) | def update(self, *args, **kwargs): method __setattr__ (line 63) | def __setattr__(self, name, value): method __setitem__ (line 68) | def __setitem__(self, name, value): class ConfigMixin (line 74) | class ConfigMixin: method register_to_config (line 97) | def register_to_config(self, **kwargs): method save_config (line 120) | def save_config(self, save_directory: Union[str, os.PathLike], push_to... method from_config (line 141) | def from_config(cls, config: Union[FrozenDict, Dict[str, Any]] = None,... method get_config_dict (line 224) | def get_config_dict(cls, *args, **kwargs): method load_config (line 233) | def load_config( method _get_init_keys (line 390) | def _get_init_keys(cls): method extract_init_dict (line 394) | def extract_init_dict(cls, config_dict, **kwargs): method _dict_from_json_file (line 478) | def _dict_from_json_file(cls, json_file: Union[str, os.PathLike]): method __repr__ (line 483) | def __repr__(self): method config (line 487) | def config(self) -> Dict[str, Any]: method to_json_string (line 496) | def to_json_string(self) -> str: method to_json_file (line 517) | def to_json_file(self, json_file_path: Union[str, os.PathLike]): function register_to_config (line 529) | def register_to_config(init): function flax_register_to_config (line 574) | def flax_register_to_config(cls): FILE: src/diffusers/dependency_versions_check.py function dep_version_check (line 46) | def dep_version_check(pkg, hint=None): FILE: src/diffusers/experimental/rl/value_guided_sampling.py class ValueGuidedRLPipeline (line 25) | class ValueGuidedRLPipeline(DiffusionPipeline): method __init__ (line 42) | def __init__( method normalize (line 70) | def normalize(self, x_in, key): method de_normalize (line 73) | def de_normalize(self, x_in, key): method to_torch (line 76) | def to_torch(self, x_in): method reset_x0 (line 83) | def reset_x0(self, x_in, cond, act_dim): method run_diffusion (line 88) | def run_diffusion(self, x, conditions, n_guide_steps, scale): method __call__ (line 121) | def __call__(self, obs, batch_size=64, planning_horizon=32, n_guide_st... FILE: src/diffusers/loaders.py class AttnProcsLayers (line 36) | class AttnProcsLayers(torch.nn.Module): method __init__ (line 37) | def __init__(self, state_dict: Dict[str, torch.Tensor]): class UNet2DConditionLoadersMixin (line 66) | class UNet2DConditionLoadersMixin: method load_attn_procs (line 67) | def load_attn_procs(self, pretrained_model_name_or_path_or_dict: Union... method save_attn_procs (line 224) | def save_attn_procs( FILE: src/diffusers/models/attention.py class AttentionBlock (line 33) | class AttentionBlock(nn.Module): method __init__ (line 51) | def __init__( method reshape_heads_to_batch_dim (line 77) | def reshape_heads_to_batch_dim(self, tensor): method reshape_batch_dim_to_heads (line 84) | def reshape_batch_dim_to_heads(self, tensor): method set_use_memory_efficient_attention_xformers (line 91) | def set_use_memory_efficient_attention_xformers( method forward (line 121) | def forward(self, hidden_states): class BasicTransformerBlock (line 177) | class BasicTransformerBlock(nn.Module): method __init__ (line 194) | def __init__( method forward (line 271) | def forward( class FeedForward (line 331) | class FeedForward(nn.Module): method __init__ (line 344) | def __init__( method forward (line 377) | def forward(self, hidden_states): class GELU (line 383) | class GELU(nn.Module): method __init__ (line 388) | def __init__(self, dim_in: int, dim_out: int, approximate: str = "none"): method gelu (line 393) | def gelu(self, gate): method forward (line 399) | def forward(self, hidden_states): class GEGLU (line 405) | class GEGLU(nn.Module): method __init__ (line 414) | def __init__(self, dim_in: int, dim_out: int): method gelu (line 418) | def gelu(self, gate): method forward (line 424) | def forward(self, hidden_states): class ApproximateGELU (line 429) | class ApproximateGELU(nn.Module): method __init__ (line 436) | def __init__(self, dim_in: int, dim_out: int): method forward (line 440) | def forward(self, x): class AdaLayerNorm (line 445) | class AdaLayerNorm(nn.Module): method __init__ (line 450) | def __init__(self, embedding_dim, num_embeddings): method forward (line 457) | def forward(self, x, timestep): class AdaLayerNormZero (line 464) | class AdaLayerNormZero(nn.Module): method __init__ (line 469) | def __init__(self, embedding_dim, num_embeddings): method forward (line 478) | def forward(self, x, timestep, class_labels, hidden_dtype=None): class AdaGroupNorm (line 485) | class AdaGroupNorm(nn.Module): method __init__ (line 490) | def __init__( method forward (line 508) | def forward(self, x, emb): FILE: src/diffusers/models/attention_flax.py class FlaxCrossAttention (line 19) | class FlaxCrossAttention(nn.Module): method setup (line 42) | def setup(self): method reshape_heads_to_batch_dim (line 53) | def reshape_heads_to_batch_dim(self, tensor): method reshape_batch_dim_to_heads (line 61) | def reshape_batch_dim_to_heads(self, tensor): method __call__ (line 69) | def __call__(self, hidden_states, context=None, deterministic=True): class FlaxBasicTransformerBlock (line 92) | class FlaxBasicTransformerBlock(nn.Module): method setup (line 119) | def setup(self): method __call__ (line 129) | def __call__(self, hidden_states, context, deterministic=True): class FlaxTransformer2DModel (line 151) | class FlaxTransformer2DModel(nn.Module): method setup (line 182) | def setup(self): method __call__ (line 220) | def __call__(self, hidden_states, context, deterministic=True): class FlaxFeedForward (line 245) | class FlaxFeedForward(nn.Module): method setup (line 266) | def setup(self): method __call__ (line 272) | def __call__(self, hidden_states, deterministic=True): class FlaxGEGLU (line 278) | class FlaxGEGLU(nn.Module): method setup (line 295) | def setup(self): method __call__ (line 299) | def __call__(self, hidden_states, deterministic=True): FILE: src/diffusers/models/autoencoder_kl.py class AutoencoderKLOutput (line 27) | class AutoencoderKLOutput(BaseOutput): class AutoencoderKL (line 40) | class AutoencoderKL(ModelMixin, ConfigMixin): method __init__ (line 69) | def __init__( method enable_tiling (line 124) | def enable_tiling(self, use_tiling: bool = True): method disable_tiling (line 132) | def disable_tiling(self): method enable_slicing (line 139) | def enable_slicing(self): method disable_slicing (line 146) | def disable_slicing(self): method encode (line 154) | def encode(self, x: torch.FloatTensor, return_dict: bool = True) -> Au... method _decode (line 167) | def _decode(self, z: torch.FloatTensor, return_dict: bool = True) -> U... method decode (line 180) | def decode(self, z: torch.FloatTensor, return_dict: bool = True) -> Un... method blend_v (line 192) | def blend_v(self, a, b, blend_extent): method blend_h (line 197) | def blend_h(self, a, b, blend_extent): method tiled_encode (line 202) | def tiled_encode(self, x: torch.FloatTensor, return_dict: bool = True)... method tiled_decode (line 248) | def tiled_decode(self, z: torch.FloatTensor, return_dict: bool = True)... method forward (line 294) | def forward( FILE: src/diffusers/models/controlnet.py class ControlNetOutput (line 38) | class ControlNetOutput(BaseOutput): class ControlNetConditioningEmbedding (line 43) | class ControlNetConditioningEmbedding(nn.Module): method __init__ (line 53) | def __init__( method forward (line 75) | def forward(self, conditioning): class ControlNetModel (line 88) | class ControlNetModel(ModelMixin, ConfigMixin): method __init__ (line 92) | def __init__( method attn_processors (line 262) | def attn_processors(self) -> Dict[str, AttnProcessor]: method set_attn_processor (line 286) | def set_attn_processor(self, processor: Union[AttnProcessor, Dict[str,... method set_attention_slice (line 317) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 382) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 386) | def forward( function zero_module (line 503) | def zero_module(module): FILE: src/diffusers/models/cross_attention.py class CrossAttention (line 34) | class CrossAttention(nn.Module): method __init__ (line 49) | def __init__( method set_use_memory_efficient_attention_xformers (line 108) | def set_use_memory_efficient_attention_xformers( method set_attention_slice (line 173) | def set_attention_slice(self, slice_size): method set_processor (line 188) | def set_processor(self, processor: "AttnProcessor"): method forward (line 201) | def forward(self, hidden_states, encoder_hidden_states=None, attention... method batch_to_head_dim (line 213) | def batch_to_head_dim(self, tensor): method head_to_batch_dim (line 220) | def head_to_batch_dim(self, tensor): method get_attention_scores (line 227) | def get_attention_scores(self, query, key, attention_mask=None): method prepare_attention_mask (line 258) | def prepare_attention_mask(self, attention_mask, target_length, batch_... class CrossAttnProcessor (line 290) | class CrossAttnProcessor: method __call__ (line 291) | def __call__( class LoRALinearLayer (line 326) | class LoRALinearLayer(nn.Module): method __init__ (line 327) | def __init__(self, in_features, out_features, rank=4): method forward (line 339) | def forward(self, hidden_states): class LoRACrossAttnProcessor (line 349) | class LoRACrossAttnProcessor(nn.Module): method __init__ (line 350) | def __init__(self, hidden_size, cross_attention_dim=None, rank=4): method __call__ (line 362) | def __call__( class CrossAttnAddedKVProcessor (line 391) | class CrossAttnAddedKVProcessor: method __call__ (line 392) | def __call__(self, attn: CrossAttention, hidden_states, encoder_hidden... class XFormersCrossAttnProcessor (line 433) | class XFormersCrossAttnProcessor: method __init__ (line 434) | def __init__(self, attention_op: Optional[Callable] = None): method __call__ (line 437) | def __call__(self, attn: CrossAttention, hidden_states, encoder_hidden... class AttnProcessor2_0 (line 469) | class AttnProcessor2_0: method __init__ (line 470) | def __init__(self): method __call__ (line 474) | def __call__(self, attn: CrossAttention, hidden_states, encoder_hidden... class LoRAXFormersCrossAttnProcessor (line 513) | class LoRAXFormersCrossAttnProcessor(nn.Module): method __init__ (line 514) | def __init__(self, hidden_size, cross_attention_dim, rank=4, attention... method __call__ (line 527) | def __call__( class SlicedAttnProcessor (line 557) | class SlicedAttnProcessor: method __init__ (line 558) | def __init__(self, slice_size): method __call__ (line 561) | def __call__(self, attn: CrossAttention, hidden_states, encoder_hidden... class SlicedAttnAddedKVProcessor (line 609) | class SlicedAttnAddedKVProcessor: method __init__ (line 610) | def __init__(self, slice_size): method __call__ (line 613) | def __call__(self, attn: "CrossAttention", hidden_states, encoder_hidd... FILE: src/diffusers/models/dual_transformer_2d.py class DualTransformer2DModel (line 21) | class DualTransformer2DModel(nn.Module): method __init__ (line 48) | def __init__( method forward (line 97) | def forward( FILE: src/diffusers/models/embeddings.py function get_timestep_embedding (line 22) | def get_timestep_embedding( function get_2d_sincos_pos_embed (line 65) | def get_2d_sincos_pos_embed(embed_dim, grid_size, cls_token=False, extra... function get_2d_sincos_pos_embed_from_grid (line 82) | def get_2d_sincos_pos_embed_from_grid(embed_dim, grid): function get_1d_sincos_pos_embed_from_grid (line 94) | def get_1d_sincos_pos_embed_from_grid(embed_dim, pos): class PatchEmbed (line 115) | class PatchEmbed(nn.Module): method __init__ (line 118) | def __init__( method forward (line 146) | def forward(self, latent): class TimestepEmbedding (line 155) | class TimestepEmbedding(nn.Module): method __init__ (line 156) | def __init__( method forward (line 200) | def forward(self, sample, condition=None): class Timesteps (line 215) | class Timesteps(nn.Module): method __init__ (line 216) | def __init__(self, num_channels: int, flip_sin_to_cos: bool, downscale... method forward (line 222) | def forward(self, timesteps): class GaussianFourierProjection (line 232) | class GaussianFourierProjection(nn.Module): method __init__ (line 235) | def __init__( method forward (line 249) | def forward(self, x): class ImagePositionalEmbeddings (line 262) | class ImagePositionalEmbeddings(nn.Module): method __init__ (line 286) | def __init__( method forward (line 304) | def forward(self, index): class LabelEmbedding (line 327) | class LabelEmbedding(nn.Module): method __init__ (line 337) | def __init__(self, num_classes, hidden_size, dropout_prob): method token_drop (line 344) | def token_drop(self, labels, force_drop_ids=None): method forward (line 355) | def forward(self, labels, force_drop_ids=None): class CombinedTimestepLabelEmbeddings (line 363) | class CombinedTimestepLabelEmbeddings(nn.Module): method __init__ (line 364) | def __init__(self, num_classes, embedding_dim, class_dropout_prob=0.1): method forward (line 371) | def forward(self, timestep, class_labels, hidden_dtype=None): FILE: src/diffusers/models/embeddings_flax.py function get_sinusoidal_embeddings (line 20) | def get_sinusoidal_embeddings( class FlaxTimestepEmbedding (line 58) | class FlaxTimestepEmbedding(nn.Module): method __call__ (line 72) | def __call__(self, temb): class FlaxTimesteps (line 79) | class FlaxTimesteps(nn.Module): method __call__ (line 92) | def __call__(self, timesteps): FILE: src/diffusers/models/modeling_flax_pytorch_utils.py function rename_key (line 28) | def rename_key(key): function rename_key_and_reshape_tensor (line 43) | def rename_key_and_reshape_tensor(pt_tuple_key, pt_tensor, random_flax_s... function convert_pytorch_state_dict_to_flax (line 90) | def convert_pytorch_state_dict_to_flax(pt_state_dict, flax_model, init_k... FILE: src/diffusers/models/modeling_flax_utils.py class FlaxModelMixin (line 45) | class FlaxModelMixin: method _from_config (line 57) | def _from_config(cls, config, **kwargs): method _cast_floating_to (line 63) | def _cast_floating_to(self, params: Union[Dict, FrozenDict], dtype: jn... method to_bf16 (line 87) | def to_bf16(self, params: Union[Dict, FrozenDict], mask: Any = None): method to_fp32 (line 126) | def to_fp32(self, params: Union[Dict, FrozenDict], mask: Any = None): method to_fp16 (line 153) | def to_fp16(self, params: Union[Dict, FrozenDict], mask: Any = None): method init_weights (line 192) | def init_weights(self, rng: jax.random.KeyArray) -> Dict: method from_pretrained (line 196) | def from_pretrained( method save_pretrained (line 487) | def save_pretrained( FILE: src/diffusers/models/modeling_pytorch_flax_utils.py function load_flax_checkpoint_in_pytorch_model (line 37) | def load_flax_checkpoint_in_pytorch_model(pt_model, model_file): function load_flax_weights_in_pytorch_model (line 58) | def load_flax_weights_in_pytorch_model(pt_model, flax_state): FILE: src/diffusers/models/modeling_utils.py function get_parameter_device (line 65) | def get_parameter_device(parameter: torch.nn.Module): function get_parameter_dtype (line 80) | def get_parameter_dtype(parameter: torch.nn.Module): function load_state_dict (line 95) | def load_state_dict(checkpoint_file: Union[str, os.PathLike], variant: O... function _load_state_dict_into_model (line 126) | def _load_state_dict_into_model(model_to_load, state_dict): function _add_variant (line 147) | def _add_variant(weights_name: str, variant: Optional[str] = None) -> str: class ModelMixin (line 156) | class ModelMixin(torch.nn.Module): method __init__ (line 170) | def __init__(self): method is_gradient_checkpointing (line 174) | def is_gradient_checkpointing(self) -> bool: method enable_gradient_checkpointing (line 183) | def enable_gradient_checkpointing(self): method disable_gradient_checkpointing (line 194) | def disable_gradient_checkpointing(self): method set_use_memory_efficient_attention_xformers (line 204) | def set_use_memory_efficient_attention_xformers( method enable_xformers_memory_efficient_attention (line 221) | def enable_xformers_memory_efficient_attention(self, attention_op: Opt... method disable_xformers_memory_efficient_attention (line 253) | def disable_xformers_memory_efficient_attention(self): method save_pretrained (line 259) | def save_pretrained( method from_pretrained (line 320) | def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union... method _load_pretrained_model (line 646) | def _load_pretrained_model( method device (line 750) | def device(self) -> device: method dtype (line 758) | def dtype(self) -> torch.dtype: method num_parameters (line 764) | def num_parameters(self, only_trainable: bool = False, exclude_embeddi... function _get_model_file (line 793) | def _get_model_file( FILE: src/diffusers/models/prior_transformer.py class PriorTransformerOutput (line 16) | class PriorTransformerOutput(BaseOutput): class PriorTransformer (line 26) | class PriorTransformer(ModelMixin, ConfigMixin): method __init__ (line 52) | def __init__( method forward (line 107) | def forward( method post_process_latents (line 192) | def post_process_latents(self, prior_latents): FILE: src/diffusers/models/resnet.py class Upsample1D (line 11) | class Upsample1D(nn.Module): method __init__ (line 22) | def __init__(self, channels, use_conv=False, use_conv_transpose=False,... method forward (line 36) | def forward(self, x): class Downsample1D (line 49) | class Downsample1D(nn.Module): method __init__ (line 60) | def __init__(self, channels, use_conv=False, out_channels=None, paddin... method forward (line 75) | def forward(self, x): class Upsample2D (line 80) | class Upsample2D(nn.Module): method __init__ (line 91) | def __init__(self, channels, use_conv=False, use_conv_transpose=False,... method forward (line 111) | def forward(self, hidden_states, output_size=None): class Downsample2D (line 149) | class Downsample2D(nn.Module): method __init__ (line 160) | def __init__(self, channels, use_conv=False, out_channels=None, paddin... method forward (line 184) | def forward(self, hidden_states): class FirUpsample2D (line 196) | class FirUpsample2D(nn.Module): method __init__ (line 197) | def __init__(self, channels=None, out_channels=None, use_conv=False, f... method _upsample_2d (line 206) | def _upsample_2d(self, hidden_states, weight=None, kernel=None, factor... method forward (line 286) | def forward(self, hidden_states): class FirDownsample2D (line 296) | class FirDownsample2D(nn.Module): method __init__ (line 297) | def __init__(self, channels=None, out_channels=None, use_conv=False, f... method _downsample_2d (line 306) | def _downsample_2d(self, hidden_states, weight=None, kernel=None, fact... method forward (line 360) | def forward(self, hidden_states): class KDownsample2D (line 371) | class KDownsample2D(nn.Module): method __init__ (line 372) | def __init__(self, pad_mode="reflect"): method forward (line 379) | def forward(self, x): class KUpsample2D (line 387) | class KUpsample2D(nn.Module): method __init__ (line 388) | def __init__(self, pad_mode="reflect"): method forward (line 395) | def forward(self, x): class ResnetBlock2D (line 403) | class ResnetBlock2D(nn.Module): method __init__ (line 434) | def __init__( method forward (line 534) | def forward(self, input_tensor, temb): class Mish (line 585) | class Mish(torch.nn.Module): method forward (line 586) | def forward(self, hidden_states): function rearrange_dims (line 591) | def rearrange_dims(tensor): class Conv1dBlock (line 602) | class Conv1dBlock(nn.Module): method __init__ (line 607) | def __init__(self, inp_channels, out_channels, kernel_size, n_groups=8): method forward (line 614) | def forward(self, x): class ResidualTemporalBlock1D (line 624) | class ResidualTemporalBlock1D(nn.Module): method __init__ (line 625) | def __init__(self, inp_channels, out_channels, embed_dim, kernel_size=5): method forward (line 637) | def forward(self, x, t): function upsample_2d (line 653) | def upsample_2d(hidden_states, kernel=None, factor=2, gain=1): function downsample_2d (line 690) | def downsample_2d(hidden_states, kernel=None, factor=2, gain=1): function upfirdn2d_native (line 725) | def upfirdn2d_native(tensor, kernel, up=1, down=1, pad=(0, 0)): FILE: src/diffusers/models/resnet_flax.py class FlaxUpsample2D (line 19) | class FlaxUpsample2D(nn.Module): method setup (line 23) | def setup(self): method __call__ (line 32) | def __call__(self, hidden_states): class FlaxDownsample2D (line 43) | class FlaxDownsample2D(nn.Module): method setup (line 47) | def setup(self): method __call__ (line 56) | def __call__(self, hidden_states): class FlaxResnetBlock2D (line 63) | class FlaxResnetBlock2D(nn.Module): method setup (line 70) | def setup(self): method __call__ (line 106) | def __call__(self, hidden_states, temb, deterministic=True): FILE: src/diffusers/models/transformer_2d.py class Transformer2DModelOutput (line 30) | class Transformer2DModelOutput(BaseOutput): class Transformer2DModel (line 41) | class Transformer2DModel(ModelMixin, ConfigMixin): method __init__ (line 80) | def __init__( method forward (line 214) | def forward( FILE: src/diffusers/models/unet_1d.py class UNet1DOutput (line 29) | class UNet1DOutput(BaseOutput): class UNet1DModel (line 39) | class UNet1DModel(ModelMixin, ConfigMixin): method __init__ (line 70) | def __init__( method forward (line 190) | def forward( FILE: src/diffusers/models/unet_1d_blocks.py class DownResnetBlock1D (line 23) | class DownResnetBlock1D(nn.Module): method __init__ (line 24) | def __init__( method forward (line 71) | def forward(self, hidden_states, temb=None): class UpResnetBlock1D (line 89) | class UpResnetBlock1D(nn.Module): method __init__ (line 90) | def __init__( method forward (line 135) | def forward(self, hidden_states, res_hidden_states_tuple=None, temb=No... class ValueFunctionMidBlock1D (line 153) | class ValueFunctionMidBlock1D(nn.Module): method __init__ (line 154) | def __init__(self, in_channels, out_channels, embed_dim): method forward (line 165) | def forward(self, x, temb=None): class MidResTemporalBlock1D (line 173) | class MidResTemporalBlock1D(nn.Module): method __init__ (line 174) | def __init__( method forward (line 217) | def forward(self, hidden_states, temb): class OutConv1DBlock (line 230) | class OutConv1DBlock(nn.Module): method __init__ (line 231) | def __init__(self, num_groups_out, out_channels, embed_dim, act_fn): method forward (line 241) | def forward(self, hidden_states, temb=None): class OutValueFunctionBlock (line 251) | class OutValueFunctionBlock(nn.Module): method __init__ (line 252) | def __init__(self, fc_dim, embed_dim): method forward (line 262) | def forward(self, hidden_states, temb): class Downsample1d (line 291) | class Downsample1d(nn.Module): method __init__ (line 292) | def __init__(self, kernel="linear", pad_mode="reflect"): method forward (line 299) | def forward(self, hidden_states): class Upsample1d (line 307) | class Upsample1d(nn.Module): method __init__ (line 308) | def __init__(self, kernel="linear", pad_mode="reflect"): method forward (line 315) | def forward(self, hidden_states, temb=None): class SelfAttention1d (line 323) | class SelfAttention1d(nn.Module): method __init__ (line 324) | def __init__(self, in_channels, n_head=1, dropout_rate=0.0): method transpose_for_scores (line 338) | def transpose_for_scores(self, projection: torch.Tensor) -> torch.Tensor: method forward (line 344) | def forward(self, hidden_states): class ResConvBlock (line 381) | class ResConvBlock(nn.Module): method __init__ (line 382) | def __init__(self, in_channels, mid_channels, out_channels, is_last=Fa... method forward (line 399) | def forward(self, hidden_states): class UNetMidBlock1D (line 415) | class UNetMidBlock1D(nn.Module): method __init__ (line 416) | def __init__(self, mid_channels, in_channels, out_channels=None): method forward (line 444) | def forward(self, hidden_states, temb=None): class AttnDownBlock1D (line 455) | class AttnDownBlock1D(nn.Module): method __init__ (line 456) | def __init__(self, out_channels, in_channels, mid_channels=None): method forward (line 475) | def forward(self, hidden_states, temb=None): class DownBlock1D (line 485) | class DownBlock1D(nn.Module): method __init__ (line 486) | def __init__(self, out_channels, in_channels, mid_channels=None): method forward (line 499) | def forward(self, hidden_states, temb=None): class DownBlock1DNoSkip (line 508) | class DownBlock1DNoSkip(nn.Module): method __init__ (line 509) | def __init__(self, out_channels, in_channels, mid_channels=None): method forward (line 521) | def forward(self, hidden_states, temb=None): class AttnUpBlock1D (line 529) | class AttnUpBlock1D(nn.Module): method __init__ (line 530) | def __init__(self, in_channels, out_channels, mid_channels=None): method forward (line 549) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None): class UpBlock1D (line 562) | class UpBlock1D(nn.Module): method __init__ (line 563) | def __init__(self, in_channels, out_channels, mid_channels=None): method forward (line 576) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None): class UpBlock1DNoSkip (line 588) | class UpBlock1DNoSkip(nn.Module): method __init__ (line 589) | def __init__(self, in_channels, out_channels, mid_channels=None): method forward (line 601) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None): function get_down_block (line 611) | def get_down_block(down_block_type, num_layers, in_channels, out_channel... function get_up_block (line 629) | def get_up_block(up_block_type, num_layers, in_channels, out_channels, t... function get_mid_block (line 647) | def get_mid_block(mid_block_type, num_layers, in_channels, mid_channels,... function get_out_block (line 663) | def get_out_block(*, out_block_type, num_groups_out, embed_dim, out_chan... FILE: src/diffusers/models/unet_2d.py class UNet2DOutput (line 28) | class UNet2DOutput(BaseOutput): class UNet2DModel (line 38) | class UNet2DModel(ModelMixin, ConfigMixin): method __init__ (line 81) | def __init__( method forward (line 217) | def forward( FILE: src/diffusers/models/unet_2d_blocks.py function get_down_block (line 27) | def get_down_block( function get_up_block (line 199) | def get_up_block( class UNetMidBlock2D (line 371) | class UNetMidBlock2D(nn.Module): method __init__ (line 372) | def __init__( method forward (line 440) | def forward(self, hidden_states, temb=None): class UNetMidBlock2DCrossAttn (line 450) | class UNetMidBlock2DCrossAttn(nn.Module): method __init__ (line 451) | def __init__( method forward (line 535) | def forward( class UNetMidBlock2DSimpleCrossAttn (line 550) | class UNetMidBlock2DSimpleCrossAttn(nn.Module): method __init__ (line 551) | def __init__( method forward (line 624) | def forward( class AttnDownBlock2D (line 644) | class AttnDownBlock2D(nn.Module): method __init__ (line 645) | def __init__( method forward (line 706) | def forward(self, hidden_states, temb=None): class CrossAttnDownBlock2D (line 723) | class CrossAttnDownBlock2D(nn.Module): method __init__ (line 724) | def __init__( method forward (line 810) | def forward( class DownBlock2D (line 854) | class DownBlock2D(nn.Module): method __init__ (line 855) | def __init__( method forward (line 906) | def forward(self, hidden_states, temb=None): class DownEncoderBlock2D (line 933) | class DownEncoderBlock2D(nn.Module): method __init__ (line 934) | def __init__( method forward (line 982) | def forward(self, hidden_states): class AttnDownEncoderBlock2D (line 993) | class AttnDownEncoderBlock2D(nn.Module): method __init__ (line 994) | def __init__( method forward (line 1054) | def forward(self, hidden_states): class AttnSkipDownBlock2D (line 1066) | class AttnSkipDownBlock2D(nn.Module): method __init__ (line 1067) | def __init__( method forward (line 1136) | def forward(self, hidden_states, temb=None, skip_sample=None): class SkipDownBlock2D (line 1156) | class SkipDownBlock2D(nn.Module): method __init__ (line 1157) | def __init__( method forward (line 1216) | def forward(self, hidden_states, temb=None, skip_sample=None): class ResnetDownsampleBlock2D (line 1235) | class ResnetDownsampleBlock2D(nn.Module): method __init__ (line 1236) | def __init__( method forward (line 1296) | def forward(self, hidden_states, temb=None): class SimpleCrossAttnDownBlock2D (line 1323) | class SimpleCrossAttnDownBlock2D(nn.Module): method __init__ (line 1324) | def __init__( method forward (line 1406) | def forward( class KDownBlock2D (line 1435) | class KDownBlock2D(nn.Module): method __init__ (line 1436) | def __init__( method forward (line 1481) | def forward(self, hidden_states, temb=None): class KCrossAttnDownBlock2D (line 1506) | class KCrossAttnDownBlock2D(nn.Module): method __init__ (line 1507) | def __init__( method forward (line 1571) | def forward( class AttnUpBlock2D (line 1618) | class AttnUpBlock2D(nn.Module): method __init__ (line 1619) | def __init__( method forward (line 1676) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None): class CrossAttnUpBlock2D (line 1693) | class CrossAttnUpBlock2D(nn.Module): method __init__ (line 1694) | def __init__( method forward (line 1776) | def forward( class UpBlock2D (line 1826) | class UpBlock2D(nn.Module): method __init__ (line 1827) | def __init__( method forward (line 1874) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class UpDecoderBlock2D (line 1900) | class UpDecoderBlock2D(nn.Module): method __init__ (line 1901) | def __init__( method forward (line 1943) | def forward(self, hidden_states): class AttnUpDecoderBlock2D (line 1954) | class AttnUpDecoderBlock2D(nn.Module): method __init__ (line 1955) | def __init__( method forward (line 2009) | def forward(self, hidden_states): class AttnSkipUpBlock2D (line 2021) | class AttnSkipUpBlock2D(nn.Module): method __init__ (line 2022) | def __init__( method forward (line 2101) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, s... class SkipUpBlock2D (line 2129) | class SkipUpBlock2D(nn.Module): method __init__ (line 2130) | def __init__( method forward (line 2198) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, s... class ResnetUpsampleBlock2D (line 2224) | class ResnetUpsampleBlock2D(nn.Module): method __init__ (line 2225) | def __init__( method forward (line 2288) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class SimpleCrossAttnUpBlock2D (line 2314) | class SimpleCrossAttnUpBlock2D(nn.Module): method __init__ (line 2315) | def __init__( method forward (line 2399) | def forward( class KUpBlock2D (line 2434) | class KUpBlock2D(nn.Module): method __init__ (line 2435) | def __init__( method forward (line 2482) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class KCrossAttnUpBlock2D (line 2507) | class KCrossAttnUpBlock2D(nn.Module): method __init__ (line 2508) | def __init__( method forward (line 2591) | def forward( class KAttentionBlock (line 2643) | class KAttentionBlock(nn.Module): method __init__ (line 2660) | def __init__( method _to_3d (line 2703) | def _to_3d(self, hidden_states, height, weight): method _to_4d (line 2706) | def _to_4d(self, hidden_states, height, weight): method forward (line 2709) | def forward( FILE: src/diffusers/models/unet_2d_blocks_flax.py class FlaxCrossAttnDownBlock2D (line 22) | class FlaxCrossAttnDownBlock2D(nn.Module): method setup (line 53) | def setup(self): method __call__ (line 85) | def __call__(self, hidden_states, temb, encoder_hidden_states, determi... class FlaxDownBlock2D (line 100) | class FlaxDownBlock2D(nn.Module): method setup (line 125) | def setup(self): method __call__ (line 143) | def __call__(self, hidden_states, temb, deterministic=True): class FlaxCrossAttnUpBlock2D (line 157) | class FlaxCrossAttnUpBlock2D(nn.Module): method setup (line 189) | def setup(self): method __call__ (line 222) | def __call__(self, hidden_states, res_hidden_states_tuple, temb, encod... class FlaxUpBlock2D (line 238) | class FlaxUpBlock2D(nn.Module): method setup (line 266) | def setup(self): method __call__ (line 286) | def __call__(self, hidden_states, res_hidden_states_tuple, temb, deter... class FlaxUNetMidBlock2DCrossAttn (line 301) | class FlaxUNetMidBlock2DCrossAttn(nn.Module): method setup (line 324) | def setup(self): method __call__ (line 359) | def __call__(self, hidden_states, temb, encoder_hidden_states, determi... FILE: src/diffusers/models/unet_2d_condition.py class UNet2DConditionOutput (line 43) | class UNet2DConditionOutput(BaseOutput): class UNet2DConditionModel (line 53) | class UNet2DConditionModel(ModelMixin, ConfigMixin, UNet2DConditionLoade... method __init__ (line 113) | def __init__( method attn_processors (line 364) | def attn_processors(self) -> Dict[str, AttnProcessor]: method set_attn_processor (line 387) | def set_attn_processor(self, processor: Union[AttnProcessor, Dict[str,... method set_attention_slice (line 417) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 482) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 486) | def forward( FILE: src/diffusers/models/unet_2d_condition_flax.py class FlaxUNet2DConditionOutput (line 36) | class FlaxUNet2DConditionOutput(BaseOutput): class FlaxUNet2DConditionModel (line 47) | class FlaxUNet2DConditionModel(nn.Module, FlaxModelMixin, ConfigMixin): method init_weights (line 115) | def init_weights(self, rng: jax.random.KeyArray) -> FrozenDict: method setup (line 127) | def setup(self): method __call__ (line 247) | def __call__( FILE: src/diffusers/models/vae.py class DecoderOutput (line 26) | class DecoderOutput(BaseOutput): class Encoder (line 38) | class Encoder(nn.Module): method __init__ (line 39) | def __init__( method forward (line 99) | def forward(self, x): class Decoder (line 118) | class Decoder(nn.Module): method __init__ (line 119) | def __init__( method forward (line 179) | def forward(self, z): class VectorQuantizer (line 198) | class VectorQuantizer(nn.Module): method __init__ (line 207) | def __init__( method remap_to_used (line 236) | def remap_to_used(self, inds): method unmap_to_all (line 250) | def unmap_to_all(self, inds): method forward (line 260) | def forward(self, z): method get_codebook_entry (line 294) | def get_codebook_entry(self, indices, shape): class DiagonalGaussianDistribution (line 312) | class DiagonalGaussianDistribution(object): method __init__ (line 313) | def __init__(self, parameters, deterministic=False): method sample (line 325) | def sample(self, generator: Optional[torch.Generator] = None) -> torch... method kl (line 333) | def kl(self, other=None): method nll (line 349) | def nll(self, sample, dims=[1, 2, 3]): method mode (line 355) | def mode(self): FILE: src/diffusers/models/vae_flax.py class FlaxDecoderOutput (line 33) | class FlaxDecoderOutput(BaseOutput): class FlaxAutoencoderKLOutput (line 48) | class FlaxAutoencoderKLOutput(BaseOutput): class FlaxUpsample2D (line 61) | class FlaxUpsample2D(nn.Module): method setup (line 75) | def setup(self): method __call__ (line 84) | def __call__(self, hidden_states): class FlaxDownsample2D (line 95) | class FlaxDownsample2D(nn.Module): method setup (line 109) | def setup(self): method __call__ (line 118) | def __call__(self, hidden_states): class FlaxResnetBlock2D (line 125) | class FlaxResnetBlock2D(nn.Module): method setup (line 151) | def setup(self): method __call__ (line 185) | def __call__(self, hidden_states, deterministic=True): class FlaxAttentionBlock (line 202) | class FlaxAttentionBlock(nn.Module): method setup (line 222) | def setup(self): method transpose_for_scores (line 231) | def transpose_for_scores(self, projection): method __call__ (line 239) | def __call__(self, hidden_states): class FlaxDownEncoderBlock2D (line 274) | class FlaxDownEncoderBlock2D(nn.Module): method setup (line 302) | def setup(self): method __call__ (line 320) | def __call__(self, hidden_states, deterministic=True): class FlaxUpDecoderBlock2D (line 330) | class FlaxUpDecoderBlock2D(nn.Module): method setup (line 358) | def setup(self): method __call__ (line 376) | def __call__(self, hidden_states, deterministic=True): class FlaxUNetMidBlock2D (line 386) | class FlaxUNetMidBlock2D(nn.Module): method setup (line 411) | def setup(self): method __call__ (line 448) | def __call__(self, hidden_states, deterministic=True): class FlaxEncoder (line 457) | class FlaxEncoder(nn.Module): method setup (line 501) | def setup(self): method __call__ (line 550) | def __call__(self, sample, deterministic: bool = True): class FlaxDecoder (line 569) | class FlaxDecoder(nn.Module): method setup (line 612) | def setup(self): method __call__ (line 665) | def __call__(self, sample, deterministic: bool = True): class FlaxDiagonalGaussianDistribution (line 683) | class FlaxDiagonalGaussianDistribution(object): method __init__ (line 684) | def __init__(self, parameters, deterministic=False): method sample (line 694) | def sample(self, key): method kl (line 697) | def kl(self, other=None): method nll (line 709) | def nll(self, sample, axis=[1, 2, 3]): method mode (line 716) | def mode(self): class FlaxAutoencoderKL (line 721) | class FlaxAutoencoderKL(nn.Module, FlaxModelMixin, ConfigMixin): method setup (line 780) | def setup(self): method init_weights (line 817) | def init_weights(self, rng: jax.random.KeyArray) -> FrozenDict: method encode (line 827) | def encode(self, sample, deterministic: bool = True, return_dict: bool... method decode (line 839) | def decode(self, latents, deterministic: bool = True, return_dict: boo... method __call__ (line 853) | def __call__(self, sample, sample_posterior=False, deterministic: bool... FILE: src/diffusers/models/vq_model.py class VQEncoderOutput (line 27) | class VQEncoderOutput(BaseOutput): class VQModel (line 39) | class VQModel(ModelMixin, ConfigMixin): method __init__ (line 70) | def __init__( method encode (line 117) | def encode(self, x: torch.FloatTensor, return_dict: bool = True) -> VQ... method decode (line 126) | def decode( method forward (line 142) | def forward(self, sample: torch.FloatTensor, return_dict: bool = True)... FILE: src/diffusers/optimization.py class SchedulerType (line 30) | class SchedulerType(Enum): function get_constant_schedule (line 39) | def get_constant_schedule(optimizer: Optimizer, last_epoch: int = -1): function get_constant_schedule_with_warmup (line 55) | def get_constant_schedule_with_warmup(optimizer: Optimizer, num_warmup_s... function get_linear_schedule_with_warmup (line 80) | def get_linear_schedule_with_warmup(optimizer, num_warmup_steps, num_tra... function get_cosine_schedule_with_warmup (line 109) | def get_cosine_schedule_with_warmup( function get_cosine_with_hard_restarts_schedule_with_warmup (line 143) | def get_cosine_with_hard_restarts_schedule_with_warmup( function get_polynomial_decay_schedule_with_warmup (line 178) | def get_polynomial_decay_schedule_with_warmup( function get_scheduler (line 238) | def get_scheduler( FILE: src/diffusers/pipelines/alt_diffusion/__init__.py class AltDiffusionPipelineOutput (line 13) | class AltDiffusionPipelineOutput(BaseOutput): FILE: src/diffusers/pipelines/alt_diffusion/modeling_roberta_series.py class TransformationModelOutput (line 11) | class TransformationModelOutput(ModelOutput): class RobertaSeriesConfig (line 39) | class RobertaSeriesConfig(XLMRobertaConfig): method __init__ (line 40) | def __init__( class RobertaSeriesModelWithTransformation (line 58) | class RobertaSeriesModelWithTransformation(RobertaPreTrainedModel): method __init__ (line 64) | def __init__(self, config): method forward (line 70) | def forward( FILE: src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py class AltDiffusionPipeline (line 52) | class AltDiffusionPipeline(DiffusionPipeline): method __init__ (line 81) | def __init__( method enable_vae_slicing (line 170) | def enable_vae_slicing(self): method disable_vae_slicing (line 179) | def disable_vae_slicing(self): method enable_vae_tiling (line 186) | def enable_vae_tiling(self): method disable_vae_tiling (line 195) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 202) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 227) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 256) | def _execution_device(self): method _encode_prompt (line 273) | def _encode_prompt( method run_safety_checker (line 411) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 421) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 429) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 446) | def check_inputs( method prepare_latents (line 493) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 512) | def __call__( FILE: src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py function preprocess (line 68) | def preprocess(image): class AltDiffusionImg2ImgPipeline (line 90) | class AltDiffusionImg2ImgPipeline(DiffusionPipeline): method __init__ (line 119) | def __init__( method enable_sequential_cpu_offload (line 208) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 233) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 262) | def _execution_device(self): method _encode_prompt (line 279) | def _encode_prompt( method run_safety_checker (line 417) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 427) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 435) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 452) | def check_inputs( method get_timesteps (line 492) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 501) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 555) | def __call__( FILE: src/diffusers/pipelines/audio_diffusion/mel.py class Mel (line 37) | class Mel(ConfigMixin, SchedulerMixin): method __init__ (line 52) | def __init__( method set_resolution (line 73) | def set_resolution(self, x_res: int, y_res: int): method load_audio (line 85) | def load_audio(self, audio_file: str = None, raw_audio: np.ndarray = N... method get_number_of_slices (line 101) | def get_number_of_slices(self) -> int: method get_audio_slice (line 109) | def get_audio_slice(self, slice: int = 0) -> np.ndarray: method get_sample_rate (line 120) | def get_sample_rate(self) -> int: method audio_slice_to_image (line 128) | def audio_slice_to_image(self, slice: int) -> Image.Image: method image_to_audio (line 145) | def image_to_audio(self, image: Image.Image) -> np.ndarray: FILE: src/diffusers/pipelines/audio_diffusion/pipeline_audio_diffusion.py class AudioDiffusionPipeline (line 30) | class AudioDiffusionPipeline(DiffusionPipeline): method __init__ (line 44) | def __init__( method get_input_dims (line 54) | def get_input_dims(self) -> Tuple: method get_default_steps (line 69) | def get_default_steps(self) -> int: method __call__ (line 78) | def __call__( method encode (line 216) | def encode(self, images: List[Image.Image], steps: int = 50) -> np.nda... method slerp (line 253) | def slerp(x0: torch.Tensor, x1: torch.Tensor, alpha: float) -> torch.T... FILE: src/diffusers/pipelines/dance_diffusion/pipeline_dance_diffusion.py class DanceDiffusionPipeline (line 27) | class DanceDiffusionPipeline(DiffusionPipeline): method __init__ (line 39) | def __init__(self, unet, scheduler): method __call__ (line 44) | def __call__( FILE: src/diffusers/pipelines/ddim/pipeline_ddim.py class DDIMPipeline (line 24) | class DDIMPipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet, scheduler): method __call__ (line 45) | def __call__( FILE: src/diffusers/pipelines/ddpm/pipeline_ddpm.py class DDPMPipeline (line 24) | class DDPMPipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet, scheduler): method __call__ (line 41) | def __call__( FILE: src/diffusers/pipelines/dit/pipeline_dit.py class DiTPipeline (line 31) | class DiTPipeline(DiffusionPipeline): method __init__ (line 45) | def __init__( method get_label_ids (line 63) | def get_label_ids(self, label: Union[str, List[str]]) -> List[int]: method __call__ (line 87) | def __call__( FILE: src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py class LDMTextToImagePipeline (line 32) | class LDMTextToImagePipeline(DiffusionPipeline): method __init__ (line 51) | def __init__( method __call__ (line 64) | def __call__( class LDMBertConfig (line 220) | class LDMBertConfig(PretrainedConfig): method __init__ (line 225) | def __init__( function _expand_mask (line 267) | def _expand_mask(mask: torch.Tensor, dtype: torch.dtype, tgt_len: Option... class LDMBertAttention (line 282) | class LDMBertAttention(nn.Module): method __init__ (line 285) | def __init__( method _shape (line 309) | def _shape(self, tensor: torch.Tensor, seq_len: int, bsz: int): method forward (line 312) | def forward( class LDMBertEncoderLayer (line 426) | class LDMBertEncoderLayer(nn.Module): method __init__ (line 427) | def __init__(self, config: LDMBertConfig): method forward (line 444) | def forward( class LDMBertPreTrainedModel (line 496) | class LDMBertPreTrainedModel(PreTrainedModel): method _init_weights (line 502) | def _init_weights(self, module): method _set_gradient_checkpointing (line 513) | def _set_gradient_checkpointing(self, module, value=False): method dummy_inputs (line 518) | def dummy_inputs(self): class LDMBertEncoder (line 528) | class LDMBertEncoder(LDMBertPreTrainedModel): method __init__ (line 538) | def __init__(self, config: LDMBertConfig): method get_input_embeddings (line 556) | def get_input_embeddings(self): method set_input_embeddings (line 559) | def set_input_embeddings(self, value): method forward (line 562) | def forward( class LDMBertModel (line 695) | class LDMBertModel(LDMBertPreTrainedModel): method __init__ (line 698) | def __init__(self, config: LDMBertConfig): method forward (line 703) | def forward( FILE: src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py function preprocess (line 22) | def preprocess(image): class LDMSuperResolutionPipeline (line 32) | class LDMSuperResolutionPipeline(DiffusionPipeline): method __init__ (line 49) | def __init__( method __call__ (line 66) | def __call__( FILE: src/diffusers/pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py class LDMPipeline (line 26) | class LDMPipeline(DiffusionPipeline): method __init__ (line 39) | def __init__(self, vqvae: VQModel, unet: UNet2DModel, scheduler: DDIMS... method __call__ (line 44) | def __call__( FILE: src/diffusers/pipelines/onnx_utils.py class OnnxRuntimeModel (line 51) | class OnnxRuntimeModel: method __init__ (line 52) | def __init__(self, model=None, **kwargs): method __call__ (line 58) | def __call__(self, **kwargs): method load_model (line 63) | def load_model(path: Union[str, Path], provider=None, sess_options=None): method _save_pretrained (line 79) | def _save_pretrained(self, save_directory: Union[str, Path], file_name... method save_pretrained (line 110) | def save_pretrained( method _from_pretrained (line 133) | def _from_pretrained( method from_pretrained (line 193) | def from_pretrained( FILE: src/diffusers/pipelines/paint_by_example/image_encoder.py class PaintByExampleImageEncoder (line 25) | class PaintByExampleImageEncoder(CLIPPreTrainedModel): method __init__ (line 26) | def __init__(self, config, proj_size=768): method forward (line 38) | def forward(self, pixel_values, return_uncond_vector=False): class PaintByExampleMapper (line 50) | class PaintByExampleMapper(nn.Module): method __init__ (line 51) | def __init__(self, config): method forward (line 63) | def forward(self, hidden_states): FILE: src/diffusers/pipelines/paint_by_example/pipeline_paint_by_example.py function prepare_mask_and_masked_image (line 37) | def prepare_mask_and_masked_image(image, mask): class PaintByExamplePipeline (line 137) | class PaintByExamplePipeline(DiffusionPipeline): method __init__ (line 168) | def __init__( method enable_sequential_cpu_offload (line 191) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 212) | def _execution_device(self): method run_safety_checker (line 230) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 241) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 259) | def decode_latents(self, latents): method check_inputs (line 268) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 291) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_mask_latents (line 309) | def prepare_mask_latents( method _encode_image (line 360) | def _encode_image(self, image, device, num_images_per_prompt, do_class... method __call__ (line 386) | def __call__( FILE: src/diffusers/pipelines/pipeline_flax_utils.py function import_flax_or_no_model (line 66) | def import_flax_or_no_model(module, class_name): class FlaxImagePipelineOutput (line 80) | class FlaxImagePipelineOutput(BaseOutput): class FlaxDiffusionPipeline (line 93) | class FlaxDiffusionPipeline(ConfigMixin): method register_modules (line 110) | def register_modules(self, **kwargs): method save_pretrained (line 143) | def save_pretrained(self, save_directory: Union[str, os.PathLike], par... method from_pretrained (line 194) | def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union... method _get_signature_keys (line 477) | def _get_signature_keys(obj): method components (line 485) | def components(self) -> Dict[str, Any]: method numpy_to_pil (line 522) | def numpy_to_pil(images): method progress_bar (line 538) | def progress_bar(self, iterable): method set_progress_bar_config (line 548) | def set_progress_bar_config(self, **kwargs): FILE: src/diffusers/pipelines/pipeline_utils.py class ImagePipelineOutput (line 109) | class ImagePipelineOutput(BaseOutput): class AudioPipelineOutput (line 123) | class AudioPipelineOutput(BaseOutput): function is_safetensors_compatible (line 136) | def is_safetensors_compatible(filenames, variant=None) -> bool: function variant_compatible_siblings (line 182) | def variant_compatible_siblings(info, variant=None) -> Union[List[os.Pat... class DiffusionPipeline (line 223) | class DiffusionPipeline(ConfigMixin): method register_modules (line 243) | def register_modules(self, **kwargs): method save_pretrained (line 276) | def save_pretrained( method to (line 345) | def to(self, torch_device: Optional[Union[str, torch.device]] = None, ... method device (line 400) | def device(self) -> torch.device: method from_pretrained (line 413) | def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union... method _get_signature_keys (line 972) | def _get_signature_keys(obj): method components (line 980) | def components(self) -> Dict[str, Any]: method numpy_to_pil (line 1017) | def numpy_to_pil(images): method progress_bar (line 1032) | def progress_bar(self, iterable=None, total=None): method set_progress_bar_config (line 1047) | def set_progress_bar_config(self, **kwargs): method enable_xformers_memory_efficient_attention (line 1050) | def enable_xformers_memory_efficient_attention(self, attention_op: Opt... method disable_xformers_memory_efficient_attention (line 1082) | def disable_xformers_memory_efficient_attention(self): method set_use_memory_efficient_attention_xformers (line 1088) | def set_use_memory_efficient_attention_xformers( method enable_attention_slicing (line 1107) | def enable_attention_slicing(self, slice_size: Optional[Union[str, int... method disable_attention_slicing (line 1123) | def disable_attention_slicing(self): method set_attention_slice (line 1131) | def set_attention_slice(self, slice_size: Optional[int]): FILE: src/diffusers/pipelines/pndm/pipeline_pndm.py class PNDMPipeline (line 26) | class PNDMPipeline(DiffusionPipeline): method __init__ (line 40) | def __init__(self, unet: UNet2DModel, scheduler: PNDMScheduler): method __call__ (line 48) | def __call__( FILE: src/diffusers/pipelines/repaint/pipeline_repaint.py function _preprocess_image (line 32) | def _preprocess_image(image: Union[List, PIL.Image.Image, torch.Tensor]): function _preprocess_mask (line 53) | def _preprocess_mask(mask: Union[List, PIL.Image.Image, torch.Tensor]): class RePaintPipeline (line 73) | class RePaintPipeline(DiffusionPipeline): method __init__ (line 77) | def __init__(self, unet, scheduler): method __call__ (line 82) | def __call__( FILE: src/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py class ScoreSdeVePipeline (line 25) | class ScoreSdeVePipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet: UNet2DModel, scheduler: DiffusionPipeline): method __call__ (line 41) | def __call__( FILE: src/diffusers/pipelines/semantic_stable_diffusion/__init__.py class SemanticStableDiffusionPipelineOutput (line 13) | class SemanticStableDiffusionPipelineOutput(BaseOutput): FILE: src/diffusers/pipelines/semantic_stable_diffusion/pipeline_semantic_stable_diffusion.py class SemanticStableDiffusionPipeline (line 61) | class SemanticStableDiffusionPipeline(DiffusionPipeline): method __init__ (line 93) | def __init__( method decode_latents (line 135) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 144) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 162) | def check_inputs( method prepare_latents (line 210) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 228) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/__init__.py class StableDiffusionPipelineOutput (line 22) | class StableDiffusionPipelineOutput(BaseOutput): class FlaxStableDiffusionPipelineOutput (line 112) | class FlaxStableDiffusionPipelineOutput(BaseOutput): FILE: src/diffusers/pipelines/stable_diffusion/convert_from_ckpt.py function shave_segments (line 67) | def shave_segments(path, n_shave_prefix_segments=1): function renew_resnet_paths (line 77) | def renew_resnet_paths(old_list, n_shave_prefix_segments=0): function renew_vae_resnet_paths (line 99) | def renew_vae_resnet_paths(old_list, n_shave_prefix_segments=0): function renew_attention_paths (line 115) | def renew_attention_paths(old_list, n_shave_prefix_segments=0): function renew_vae_attention_paths (line 136) | def renew_vae_attention_paths(old_list, n_shave_prefix_segments=0): function assign_to_checkpoint (line 166) | def assign_to_checkpoint( function conv_attn_to_linear (line 217) | def conv_attn_to_linear(checkpoint): function create_unet_diffusers_config (line 229) | def create_unet_diffusers_config(original_config, image_size: int, contr... function create_vae_diffusers_config (line 298) | def create_vae_diffusers_config(original_config, image_size: int): function create_diffusers_schedular (line 322) | def create_diffusers_schedular(original_config): function create_ldm_bert_config (line 332) | def create_ldm_bert_config(original_config): function convert_ldm_unet_checkpoint (line 342) | def convert_ldm_unet_checkpoint(checkpoint, config, path=None, extract_e... function convert_ldm_vae_checkpoint (line 573) | def convert_ldm_vae_checkpoint(checkpoint, config): function convert_ldm_bert_checkpoint (line 680) | def convert_ldm_bert_checkpoint(checkpoint, config): function convert_ldm_clip_checkpoint (line 730) | def convert_ldm_clip_checkpoint(checkpoint): function convert_paint_by_example_checkpoint (line 770) | def convert_paint_by_example_checkpoint(checkpoint): function convert_open_clip_checkpoint (line 837) | def convert_open_clip_checkpoint(checkpoint): function stable_unclip_image_encoder (line 880) | def stable_unclip_image_encoder(original_config): function stable_unclip_image_noising_components (line 913) | def stable_unclip_image_noising_components( function load_pipeline_from_original_stable_diffusion_ckpt (line 958) | def load_pipeline_from_original_stable_diffusion_ckpt( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_cycle_diffusion.py function preprocess (line 39) | def preprocess(image): function posterior_sample (line 60) | def posterior_sample(scheduler, latents, timestep, clean_latents, genera... function compute_noise (line 87) | def compute_noise(scheduler, prev_latents, latents, timestep, noise_pred... class CycleDiffusionPipeline (line 121) | class CycleDiffusionPipeline(DiffusionPipeline): method __init__ (line 150) | def __init__( method enable_sequential_cpu_offload (line 225) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 251) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 281) | def _execution_device(self): method _encode_prompt (line 299) | def _encode_prompt( method check_inputs (line 438) | def check_inputs( method prepare_extra_step_kwargs (line 479) | def prepare_extra_step_kwargs(self, generator, eta): method run_safety_checker (line 497) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 508) | def decode_latents(self, latents): method get_timesteps (line 517) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 526) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 576) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion.py class FlaxStableDiffusionPipeline (line 48) | class FlaxStableDiffusionPipeline(FlaxDiffusionPipeline): method __init__ (line 77) | def __init__( method prepare_inputs (line 135) | def prepare_inputs(self, prompt: Union[str, List[str]]): method _get_has_nsfw_concepts (line 148) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 152) | def _run_safety_checker(self, images, safety_model_params, jit=False): method _generate (line 182) | def _generate( method __call__ (line 275) | def __call__( function _p_generate (line 397) | def _p_generate( function _p_get_has_nsfw_concepts (line 423) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 427) | def unshard(x: jnp.ndarray): FILE: src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_img2img.py class FlaxStableDiffusionImg2ImgPipeline (line 47) | class FlaxStableDiffusionImg2ImgPipeline(FlaxDiffusionPipeline): method __init__ (line 76) | def __init__( method prepare_inputs (line 113) | def prepare_inputs(self, prompt: Union[str, List[str]], image: Union[I... method _get_has_nsfw_concepts (line 134) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 138) | def _run_safety_checker(self, images, safety_model_params, jit=False): method get_timestep_start (line 168) | def get_timestep_start(self, num_inference_steps, strength): method _generate (line 176) | def _generate( method __call__ (line 280) | def __call__( function _p_generate (line 419) | def _p_generate( function _p_get_has_nsfw_concepts (line 449) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 453) | def unshard(x: jnp.ndarray): function preprocess (line 460) | def preprocess(image, dtype): FILE: src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_inpaint.py class FlaxStableDiffusionInpaintPipeline (line 48) | class FlaxStableDiffusionInpaintPipeline(FlaxDiffusionPipeline): method __init__ (line 77) | def __init__( method prepare_inputs (line 135) | def prepare_inputs( method _get_has_nsfw_concepts (line 174) | def _get_has_nsfw_concepts(self, features, params): method _run_safety_checker (line 178) | def _run_safety_checker(self, images, safety_model_params, jit=False): method _generate (line 208) | def _generate( method __call__ (line 335) | def __call__( function _p_generate (line 466) | def _p_generate( function _p_get_has_nsfw_concepts (line 496) | def _p_get_has_nsfw_concepts(pipe, features, params): function unshard (line 500) | def unshard(x: jnp.ndarray): function preprocess_image (line 507) | def preprocess_image(image, dtype): function preprocess_mask (line 516) | def preprocess_mask(mask, dtype): FILE: src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py class OnnxStableDiffusionPipeline (line 33) | class OnnxStableDiffusionPipeline(DiffusionPipeline): method __init__ (line 45) | def __init__( method _encode_prompt (line 114) | def _encode_prompt(self, prompt, num_images_per_prompt, do_classifier_... method __call__ (line 191) | def __call__( class StableDiffusionOnnxPipeline (line 326) | class StableDiffusionOnnxPipeline(OnnxStableDiffusionPipeline): method __init__ (line 327) | def __init__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py function preprocess (line 35) | def preprocess(image): class OnnxStableDiffusionImg2ImgPipeline (line 56) | class OnnxStableDiffusionImg2ImgPipeline(DiffusionPipeline): method __init__ (line 94) | def __init__( method _encode_prompt (line 164) | def _encode_prompt(self, prompt, num_images_per_prompt, do_classifier_... method __call__ (line 241) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint.py function prepare_mask_and_masked_image (line 38) | def prepare_mask_and_masked_image(image, mask, latents_shape): class OnnxStableDiffusionInpaintPipeline (line 56) | class OnnxStableDiffusionInpaintPipeline(DiffusionPipeline): method __init__ (line 94) | def __init__( method _encode_prompt (line 165) | def _encode_prompt(self, prompt, num_images_per_prompt, do_classifier_... method __call__ (line 243) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint_legacy.py function preprocess (line 20) | def preprocess(image): function preprocess_mask (line 29) | def preprocess_mask(mask, scale_factor=8): class OnnxStableDiffusionInpaintPipelineLegacy (line 41) | class OnnxStableDiffusionInpaintPipelineLegacy(DiffusionPipeline): method __init__ (line 80) | def __init__( method _encode_prompt (line 150) | def _encode_prompt(self, prompt, num_images_per_prompt, do_classifier_... method __call__ (line 227) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py class StableDiffusionPipeline (line 55) | class StableDiffusionPipeline(DiffusionPipeline): method __init__ (line 84) | def __init__( method enable_vae_slicing (line 173) | def enable_vae_slicing(self): method disable_vae_slicing (line 182) | def disable_vae_slicing(self): method enable_vae_tiling (line 189) | def enable_vae_tiling(self): method disable_vae_tiling (line 198) | def disable_vae_tiling(self): method enable_sequential_cpu_offload (line 205) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 230) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 259) | def _execution_device(self): method _encode_prompt (line 276) | def _encode_prompt( method run_safety_checker (line 414) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 424) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 432) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 449) | def check_inputs( method prepare_latents (line 496) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 515) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py class AttentionStore (line 71) | class AttentionStore: method get_empty_store (line 73) | def get_empty_store(): method __call__ (line 76) | def __call__(self, attn, is_cross: bool, place_in_unet: str): method between_steps (line 86) | def between_steps(self): method get_average_attention (line 90) | def get_average_attention(self): method aggregate_attention (line 94) | def aggregate_attention(self, from_where: List[str]) -> torch.Tensor: method reset (line 106) | def reset(self): method __init__ (line 111) | def __init__(self, attn_res=16): class AttendExciteCrossAttnProcessor (line 124) | class AttendExciteCrossAttnProcessor: method __init__ (line 125) | def __init__(self, attnstore, place_in_unet): method __call__ (line 130) | def __call__(self, attn: CrossAttention, hidden_states, encoder_hidden... class StableDiffusionAttendAndExcitePipeline (line 162) | class StableDiffusionAttendAndExcitePipeline(DiffusionPipeline): method __init__ (line 191) | def __init__( method enable_vae_slicing (line 233) | def enable_vae_slicing(self): method disable_vae_slicing (line 243) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 251) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 278) | def _execution_device(self): method _encode_prompt (line 296) | def _encode_prompt( method run_safety_checker (line 435) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 446) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 455) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 472) | def check_inputs( method prepare_latents (line 546) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method _compute_max_attention_per_index (line 564) | def _compute_max_attention_per_index( method _aggregate_and_get_max_attention_per_token (line 586) | def _aggregate_and_get_max_attention_per_token( method _compute_loss (line 601) | def _compute_loss(max_attention_per_index: List[torch.Tensor]) -> torc... method _update_latent (line 608) | def _update_latent(latents: torch.Tensor, loss: torch.Tensor, step_siz... method _perform_iterative_refinement_step (line 614) | def _perform_iterative_refinement_step( method register_attention_control (line 668) | def register_attention_control(self): method get_indices (line 689) | def get_indices(self, prompt: str) -> Dict[str, int]: method __call__ (line 697) | def __call__( class GaussianSmoothing (line 978) | class GaussianSmoothing(torch.nn.Module): method __init__ (line 991) | def __init__( method forward (line 1032) | def forward(self, input): FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_controlnet.py class StableDiffusionControlNetPipeline (line 88) | class StableDiffusionControlNetPipeline(DiffusionPipeline): method __init__ (line 119) | def __init__( method enable_vae_slicing (line 163) | def enable_vae_slicing(self): method disable_vae_slicing (line 173) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 180) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 201) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 231) | def _execution_device(self): method _encode_prompt (line 249) | def _encode_prompt( method run_safety_checker (line 388) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 399) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 408) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 425) | def check_inputs( method prepare_image (line 504) | def prepare_image(self, image, width, height, batch_size, num_images_p... method prepare_latents (line 535) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method _default_height_width (line 552) | def _default_height_width(self, height, width, image): method __call__ (line 576) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py function preprocess (line 36) | def preprocess(image): class StableDiffusionDepth2ImgPipeline (line 57) | class StableDiffusionDepth2ImgPipeline(DiffusionPipeline): method __init__ (line 80) | def __init__( method enable_sequential_cpu_offload (line 124) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 143) | def _execution_device(self): method _encode_prompt (line 161) | def _encode_prompt( method run_safety_checker (line 300) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 311) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 320) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 338) | def check_inputs( method get_timesteps (line 379) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 389) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method prepare_depth_map (line 441) | def prepare_depth_map(self, image, depth_map, batch_size, do_classifie... method __call__ (line 483) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py class StableDiffusionImageVariationPipeline (line 35) | class StableDiffusionImageVariationPipeline(DiffusionPipeline): method __init__ (line 63) | def __init__( method enable_sequential_cpu_offload (line 123) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 142) | def _execution_device(self): method _encode_image (line 159) | def _encode_image(self, image, device, num_images_per_prompt, do_class... method run_safety_checker (line 185) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 196) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 205) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 222) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 245) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 263) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py function preprocess (line 72) | def preprocess(image): class StableDiffusionImg2ImgPipeline (line 93) | class StableDiffusionImg2ImgPipeline(DiffusionPipeline): method __init__ (line 123) | def __init__( method enable_sequential_cpu_offload (line 213) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 239) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 269) | def _execution_device(self): method _encode_prompt (line 287) | def _encode_prompt( method run_safety_checker (line 426) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 437) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 446) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 463) | def check_inputs( method get_timesteps (line 503) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 512) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 566) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py function prepare_mask_and_masked_image (line 36) | def prepare_mask_and_masked_image(image, mask): class StableDiffusionInpaintPipeline (line 140) | class StableDiffusionInpaintPipeline(DiffusionPipeline): method __init__ (line 169) | def __init__( method enable_sequential_cpu_offload (line 260) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 286) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 316) | def _execution_device(self): method _encode_prompt (line 334) | def _encode_prompt( method run_safety_checker (line 473) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 484) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 502) | def decode_latents(self, latents): method check_inputs (line 511) | def check_inputs( method prepare_latents (line 559) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_mask_latents (line 576) | def prepare_mask_latents( method __call__ (line 628) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py function preprocess_image (line 43) | def preprocess_image(image): function preprocess_mask (line 53) | def preprocess_mask(mask, scale_factor=8): class StableDiffusionInpaintPipelineLegacy (line 84) | class StableDiffusionInpaintPipelineLegacy(DiffusionPipeline): method __init__ (line 114) | def __init__( method enable_sequential_cpu_offload (line 204) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 230) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 260) | def _execution_device(self): method _encode_prompt (line 278) | def _encode_prompt( method run_safety_checker (line 417) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 428) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 437) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 455) | def check_inputs( method get_timesteps (line 496) | def get_timesteps(self, num_inference_steps, strength, device): method prepare_latents (line 505) | def prepare_latents(self, image, timestep, batch_size, num_images_per_... method __call__ (line 522) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py function preprocess (line 42) | def preprocess(image): class StableDiffusionInstructPix2PixPipeline (line 63) | class StableDiffusionInstructPix2PixPipeline(DiffusionPipeline): method __init__ (line 92) | def __init__( method __call__ (line 134) | def __call__( method enable_sequential_cpu_offload (line 399) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 425) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 455) | def _execution_device(self): method _encode_prompt (line 472) | def _encode_prompt( method run_safety_checker (line 612) | def run_safety_checker(self, image, device, dtype): method prepare_extra_step_kwargs (line 623) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 641) | def decode_latents(self, latents): method check_inputs (line 649) | def check_inputs( method prepare_latents (line 687) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method prepare_image_latents (line 704) | def prepare_image_latents( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py class ModelWrapper (line 30) | class ModelWrapper: method __init__ (line 31) | def __init__(self, model, alphas_cumprod): method apply_model (line 35) | def apply_model(self, *args, **kwargs): class StableDiffusionKDiffusionPipeline (line 44) | class StableDiffusionKDiffusionPipeline(DiffusionPipeline): method __init__ (line 79) | def __init__( method set_scheduler (line 119) | def set_scheduler(self, scheduler_type: str): method enable_sequential_cpu_offload (line 125) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 151) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 181) | def _execution_device(self): method _encode_prompt (line 199) | def _encode_prompt( method run_safety_checker (line 338) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 349) | def decode_latents(self, latents): method check_inputs (line 357) | def check_inputs(self, prompt, height, width, callback_steps): method prepare_latents (line 372) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 385) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py function preprocess (line 33) | def preprocess(image): class StableDiffusionLatentUpscalePipeline (line 54) | class StableDiffusionLatentUpscalePipeline(DiffusionPipeline): method __init__ (line 77) | def __init__( method enable_sequential_cpu_offload (line 95) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 114) | def _execution_device(self): method _encode_prompt (line 131) | def _encode_prompt(self, prompt, device, do_classifier_free_guidance, ... method decode_latents (line 222) | def decode_latents(self, latents): method check_inputs (line 230) | def check_inputs(self, prompt, image, callback_steps): method prepare_latents (line 268) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 282) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_panorama.py class StableDiffusionPanoramaPipeline (line 50) | class StableDiffusionPanoramaPipeline(DiffusionPipeline): method __init__ (line 83) | def __init__( method enable_vae_slicing (line 128) | def enable_vae_slicing(self): method disable_vae_slicing (line 138) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 146) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 173) | def _execution_device(self): method _encode_prompt (line 191) | def _encode_prompt( method run_safety_checker (line 330) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 341) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 350) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 368) | def check_inputs( method prepare_latents (line 416) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method get_views (line 433) | def get_views(self, panorama_height, panorama_width, window_size=64, s... method __call__ (line 451) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py class Pix2PixInversionPipelineOutput (line 53) | class Pix2PixInversionPipelineOutput(BaseOutput): function preprocess (line 173) | def preprocess(image): function prepare_unet (line 194) | def prepare_unet(unet: UNet2DConditionModel): class Pix2PixZeroL2Loss (line 211) | class Pix2PixZeroL2Loss: method __init__ (line 212) | def __init__(self): method compute_loss (line 215) | def compute_loss(self, predictions, targets): class Pix2PixZeroCrossAttnProcessor (line 219) | class Pix2PixZeroCrossAttnProcessor: method __init__ (line 223) | def __init__(self, is_pix2pix_zero=False): method __call__ (line 228) | def __call__( class StableDiffusionPix2PixZeroPipeline (line 274) | class StableDiffusionPix2PixZeroPipeline(DiffusionPipeline): method __init__ (line 312) | def __init__( method enable_sequential_cpu_offload (line 360) | def enable_sequential_cpu_offload(self, gpu_id=0): method enable_model_cpu_offload (line 385) | def enable_model_cpu_offload(self, gpu_id=0): method _execution_device (line 411) | def _execution_device(self): method _encode_prompt (line 429) | def _encode_prompt( method run_safety_checker (line 568) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 579) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 588) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 605) | def check_inputs( method prepare_latents (line 637) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method generate_caption (line 655) | def generate_caption(self, images): method construct_direction (line 674) | def construct_direction(self, embs_source: torch.Tensor, embs_target: ... method get_embeds (line 679) | def get_embeds(self, prompt: List[str], batch_size: int = 16) -> torch... method prepare_image_latents (line 698) | def prepare_image_latents(self, image, batch_size, dtype, device, gene... method auto_corr_loss (line 733) | def auto_corr_loss(self, hidden_states, generator=None): method kl_divergence (line 753) | def kl_divergence(self, hidden_states): method __call__ (line 760) | def __call__( method invert (line 1037) | def invert( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_sag.py class CrossAttnStoreProcessor (line 51) | class CrossAttnStoreProcessor: method __init__ (line 52) | def __init__(self): method __call__ (line 55) | def __call__( class StableDiffusionSAGPipeline (line 91) | class StableDiffusionSAGPipeline(DiffusionPipeline): method __init__ (line 120) | def __init__( method enable_vae_slicing (line 146) | def enable_vae_slicing(self): method disable_vae_slicing (line 156) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 164) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 191) | def _execution_device(self): method _encode_prompt (line 209) | def _encode_prompt( method run_safety_checker (line 348) | def run_safety_checker(self, image, device, dtype): method decode_latents (line 359) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 368) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 386) | def check_inputs( method prepare_latents (line 434) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 453) | def __call__( method sag_masking (line 683) | def sag_masking(self, original_latents, attn_map, t, eps): method pred_x0 (line 711) | def pred_x0(self, sample, model_output, timestep): method pred_epsilon (line 731) | def pred_epsilon(self, sample, model_output, timestep): function gaussian_blur_2d (line 751) | def gaussian_blur_2d(img, kernel_size, sigma): FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py function preprocess (line 32) | def preprocess(image): class StableDiffusionUpscalePipeline (line 53) | class StableDiffusionUpscalePipeline(DiffusionPipeline): method __init__ (line 79) | def __init__( method enable_sequential_cpu_offload (line 117) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 136) | def _execution_device(self): method _encode_prompt (line 154) | def _encode_prompt( method prepare_extra_step_kwargs (line 293) | def prepare_extra_step_kwargs(self, generator, eta): method decode_latents (line 311) | def decode_latents(self, latents): method check_inputs (line 319) | def check_inputs(self, prompt, image, noise_level, callback_steps): method prepare_latents (line 360) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 374) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip.py class StableUnCLIPPipeline (line 50) | class StableUnCLIPPipeline(DiffusionPipeline): method __init__ (line 103) | def __init__( method enable_vae_slicing (line 140) | def enable_vae_slicing(self): method disable_vae_slicing (line 150) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 157) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 183) | def _execution_device(self): method _encode_prior_prompt (line 201) | def _encode_prior_prompt( method _encode_prompt (line 303) | def _encode_prompt( method decode_latents (line 442) | def decode_latents(self, latents): method prepare_prior_extra_step_kwargs (line 451) | def prepare_prior_extra_step_kwargs(self, generator, eta): method prepare_extra_step_kwargs (line 469) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 486) | def check_inputs( method prepare_latents (line 547) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method noise_image_embeddings (line 558) | def noise_image_embeddings( method __call__ (line 605) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py class StableUnCLIPImg2ImgPipeline (line 63) | class StableUnCLIPImg2ImgPipeline(DiffusionPipeline): method __init__ (line 109) | def __init__( method enable_vae_slicing (line 142) | def enable_vae_slicing(self): method disable_vae_slicing (line 152) | def disable_vae_slicing(self): method enable_sequential_cpu_offload (line 159) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 185) | def _execution_device(self): method _encode_prompt (line 203) | def _encode_prompt( method _encode_image (line 341) | def _encode_image( method decode_latents (line 397) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 406) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 423) | def check_inputs( method prepare_latents (line 507) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method noise_image_embeddings (line 525) | def noise_image_embeddings( method __call__ (line 572) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/safety_checker.py function cosine_distance (line 26) | def cosine_distance(image_embeds, text_embeds): class StableDiffusionSafetyChecker (line 32) | class StableDiffusionSafetyChecker(PreTrainedModel): method __init__ (line 37) | def __init__(self, config: CLIPConfig): method forward (line 50) | def forward(self, clip_input, images): method forward_onnx (line 99) | def forward_onnx(self, clip_input: torch.FloatTensor, images: torch.Fl... FILE: src/diffusers/pipelines/stable_diffusion/safety_checker_flax.py function jax_cosine_distance (line 25) | def jax_cosine_distance(emb_1, emb_2, eps=1e-12): class FlaxStableDiffusionSafetyCheckerModule (line 31) | class FlaxStableDiffusionSafetyCheckerModule(nn.Module): method setup (line 35) | def setup(self): method __call__ (line 47) | def __call__(self, clip_input): class FlaxStableDiffusionSafetyChecker (line 71) | class FlaxStableDiffusionSafetyChecker(FlaxPreTrainedModel): method __init__ (line 76) | def __init__( method init_weights (line 90) | def init_weights(self, rng: jax.random.KeyArray, input_shape: Tuple, p... method __call__ (line 101) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion/stable_unclip_image_normalizer.py class StableUnCLIPImageNormalizer (line 22) | class StableUnCLIPImageNormalizer(ModelMixin, ConfigMixin): method __init__ (line 31) | def __init__( method scale (line 40) | def scale(self, embeds): method unscale (line 44) | def unscale(self, embeds): FILE: src/diffusers/pipelines/stable_diffusion_safe/__init__.py class SafetyConfig (line 13) | class SafetyConfig(object): class StableDiffusionSafePipelineOutput (line 45) | class StableDiffusionSafePipelineOutput(BaseOutput): FILE: src/diffusers/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py class StableDiffusionPipelineSafe (line 22) | class StableDiffusionPipelineSafe(DiffusionPipeline): method __init__ (line 54) | def __init__( method safety_concept (line 150) | def safety_concept(self): method safety_concept (line 160) | def safety_concept(self, concept): method enable_sequential_cpu_offload (line 170) | def enable_sequential_cpu_offload(self): method _execution_device (line 189) | def _execution_device(self): method _encode_prompt (line 206) | def _encode_prompt( method run_safety_checker (line 341) | def run_safety_checker(self, image, device, dtype, enable_safety_guida... method decode_latents (line 365) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 374) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 392) | def check_inputs( method prepare_latents (line 440) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method perform_safety_guidance (line 457) | def perform_safety_guidance( method __call__ (line 500) | def __call__( FILE: src/diffusers/pipelines/stable_diffusion_safe/safety_checker.py function cosine_distance (line 25) | def cosine_distance(image_embeds, text_embeds): class SafeStableDiffusionSafetyChecker (line 31) | class SafeStableDiffusionSafetyChecker(PreTrainedModel): method __init__ (line 36) | def __init__(self, config: CLIPConfig): method forward (line 49) | def forward(self, clip_input, images): method forward_onnx (line 88) | def forward_onnx(self, clip_input: torch.FloatTensor, images: torch.Fl... FILE: src/diffusers/pipelines/stochastic_karras_ve/pipeline_stochastic_karras_ve.py class KarrasVePipeline (line 25) | class KarrasVePipeline(DiffusionPipeline): method __init__ (line 44) | def __init__(self, unet: UNet2DModel, scheduler: KarrasVeScheduler): method __call__ (line 49) | def __call__( FILE: src/diffusers/pipelines/unclip/pipeline_unclip.py class UnCLIPPipeline (line 34) | class UnCLIPPipeline(DiffusionPipeline): method __init__ (line 78) | def __init__( method prepare_latents (line 106) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 117) | def _encode_prompt( method enable_sequential_cpu_offload (line 208) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 234) | def _execution_device(self): method __call__ (line 252) | def __call__( FILE: src/diffusers/pipelines/unclip/pipeline_unclip_image_variation.py class UnCLIPImageVariationPipeline (line 38) | class UnCLIPImageVariationPipeline(DiffusionPipeline): method __init__ (line 84) | def __init__( method prepare_latents (line 113) | def prepare_latents(self, shape, dtype, device, generator, latents, sc... method _encode_prompt (line 124) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method _encode_image (line 187) | def _encode_image(self, image, device, num_images_per_prompt, image_em... method enable_sequential_cpu_offload (line 201) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 227) | def _execution_device(self): method __call__ (line 245) | def __call__( FILE: src/diffusers/pipelines/unclip/text_proj.py class UnCLIPTextProjModel (line 22) | class UnCLIPTextProjModel(ModelMixin, ConfigMixin): method __init__ (line 31) | def __init__( method forward (line 55) | def forward(self, *, image_embeddings, prompt_embeds, text_encoder_hid... FILE: src/diffusers/pipelines/versatile_diffusion/modeling_text_unet.py function get_down_block (line 21) | def get_down_block( function get_up_block (line 77) | def get_up_block( class UNetFlatConditionModel (line 134) | class UNetFlatConditionModel(ModelMixin, ConfigMixin): method __init__ (line 194) | def __init__( method attn_processors (line 454) | def attn_processors(self) -> Dict[str, AttnProcessor]: method set_attn_processor (line 477) | def set_attn_processor(self, processor: Union[AttnProcessor, Dict[str,... method set_attention_slice (line 507) | def set_attention_slice(self, slice_size): method _set_gradient_checkpointing (line 572) | def _set_gradient_checkpointing(self, module, value=False): method forward (line 576) | def forward( class LinearMultiDim (line 747) | class LinearMultiDim(nn.Linear): method __init__ (line 748) | def __init__(self, in_features, out_features=None, second_dim=4, *args... method forward (line 757) | def forward(self, input_tensor, *args, **kwargs): class ResnetBlockFlat (line 766) | class ResnetBlockFlat(nn.Module): method __init__ (line 767) | def __init__( method forward (line 827) | def forward(self, input_tensor, temb): class DownBlockFlat (line 861) | class DownBlockFlat(nn.Module): method __init__ (line 862) | def __init__( method forward (line 913) | def forward(self, hidden_states, temb=None): class CrossAttnDownBlockFlat (line 941) | class CrossAttnDownBlockFlat(nn.Module): method __init__ (line 942) | def __init__( method forward (line 1028) | def forward( class UpBlockFlat (line 1073) | class UpBlockFlat(nn.Module): method __init__ (line 1074) | def __init__( method forward (line 1121) | def forward(self, hidden_states, res_hidden_states_tuple, temb=None, u... class CrossAttnUpBlockFlat (line 1148) | class CrossAttnUpBlockFlat(nn.Module): method __init__ (line 1149) | def __init__( method forward (line 1231) | def forward( class UNetMidBlockFlatCrossAttn (line 1282) | class UNetMidBlockFlatCrossAttn(nn.Module): method __init__ (line 1283) | def __init__( method forward (line 1367) | def forward( class UNetMidBlockFlatSimpleCrossAttn (line 1383) | class UNetMidBlockFlatSimpleCrossAttn(nn.Module): method __init__ (line 1384) | def __init__( method forward (line 1457) | def forward( FILE: src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion.py class VersatileDiffusionPipeline (line 20) | class VersatileDiffusionPipeline(DiffusionPipeline): method __init__ (line 57) | def __init__( method image_variation (line 83) | def image_variation( method text_to_image (line 199) | def text_to_image( method dual_guided (line 311) | def dual_guided( FILE: src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py class VersatileDiffusionDualGuidedPipeline (line 39) | class VersatileDiffusionDualGuidedPipeline(DiffusionPipeline): method __init__ (line 68) | def __init__( method remove_unused_weights (line 98) | def remove_unused_weights(self): method _convert_to_dual_attention (line 101) | def _convert_to_dual_attention(self): method _revert_dual_attention (line 135) | def _revert_dual_attention(self): method enable_sequential_cpu_offload (line 148) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 167) | def _execution_device(self): method _encode_text_prompt (line 184) | def _encode_text_prompt(self, prompt, device, num_images_per_prompt, d... method _encode_image_prompt (line 275) | def _encode_image_prompt(self, prompt, device, num_images_per_prompt, ... method decode_latents (line 331) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 340) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 357) | def check_inputs(self, prompt, image, height, width, callback_steps): method prepare_latents (line 375) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method set_transformer_params (line 392) | def set_transformer_params(self, mix_ratio: float = 0.5, condition_typ... method __call__ (line 406) | def __call__( FILE: src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py class VersatileDiffusionImageVariationPipeline (line 33) | class VersatileDiffusionImageVariationPipeline(DiffusionPipeline): method __init__ (line 57) | def __init__( method enable_sequential_cpu_offload (line 75) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 94) | def _execution_device(self): method _encode_prompt (line 111) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method decode_latents (line 191) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 200) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 218) | def check_inputs(self, image, height, width, callback_steps): method prepare_latents (line 241) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 259) | def __call__( FILE: src/diffusers/pipelines/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py class VersatileDiffusionTextToImagePipeline (line 32) | class VersatileDiffusionTextToImagePipeline(DiffusionPipeline): method __init__ (line 60) | def __init__( method _swap_unet_attention_blocks (line 83) | def _swap_unet_attention_blocks(self): method remove_unused_weights (line 96) | def remove_unused_weights(self): method enable_sequential_cpu_offload (line 99) | def enable_sequential_cpu_offload(self, gpu_id=0): method _execution_device (line 118) | def _execution_device(self): method _encode_prompt (line 135) | def _encode_prompt(self, prompt, device, num_images_per_prompt, do_cla... method decode_latents (line 248) | def decode_latents(self, latents): method prepare_extra_step_kwargs (line 257) | def prepare_extra_step_kwargs(self, generator, eta): method check_inputs (line 275) | def check_inputs( method prepare_latents (line 323) | def prepare_latents(self, batch_size, num_channels_latents, height, wi... method __call__ (line 341) | def __call__( FILE: src/diffusers/pipelines/vq_diffusion/pipeline_vq_diffusion.py class LearnedClassifierFreeSamplingEmbeddings (line 30) | class LearnedClassifierFreeSamplingEmbeddings(ModelMixin, ConfigMixin): method __init__ (line 36) | def __init__(self, learnable: bool, hidden_size: Optional[int] = None,... class VQDiffusionPipeline (line 52) | class VQDiffusionPipeline(DiffusionPipeline): method __init__ (line 83) | def __init__( method _encode_prompt (line 103) | def _encode_prompt(self, prompt, num_images_per_prompt, do_classifier_... method __call__ (line 167) | def __call__( method truncate (line 310) | def truncate(self, log_p_x_0: torch.FloatTensor, truncation_rate: floa... FILE: src/diffusers/schedulers/scheduling_ddim.py class DDIMSchedulerOutput (line 32) | class DDIMSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999) -> torc... class DDIMScheduler (line 79) | class DDIMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 120) | def __init__( method scale_model_input (line 163) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method _get_variance (line 177) | def _get_variance(self, timestep, prev_timestep): method set_timesteps (line 187) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 211) | def step( method add_noise (line 329) | def add_noise( method get_velocity (line 352) | def get_velocity( method __len__ (line 372) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_ddim_flax.py class DDIMSchedulerState (line 36) | class DDIMSchedulerState: method create (line 46) | def create( class FlaxDDIMSchedulerOutput (line 62) | class FlaxDDIMSchedulerOutput(FlaxSchedulerOutput): class FlaxDDIMScheduler (line 66) | class FlaxDDIMScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 109) | def has_state(self): method __init__ (line 113) | def __init__( method create_state (line 127) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method scale_model_input (line 151) | def scale_model_input( method set_timesteps (line 165) | def set_timesteps( method _get_variance (line 187) | def _get_variance(self, state: DDIMSchedulerState, timestep, prev_time... method step (line 199) | def step( method add_noise (line 285) | def add_noise( method get_velocity (line 294) | def get_velocity( method __len__ (line 303) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_ddim_inverse.py class DDIMSchedulerOutput (line 31) | class DDIMSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 49) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999) -> torc... class DDIMInverseScheduler (line 78) | class DDIMInverseScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 117) | def __init__( method scale_model_input (line 160) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 174) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 198) | def step( method __len__ (line 226) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_ddpm.py class DDPMSchedulerOutput (line 30) | class DDPMSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 47) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class DDPMScheduler (line 76) | class DDPMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 112) | def __init__( method scale_model_input (line 156) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 170) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _get_variance (line 192) | def _get_variance(self, t, predicted_variance=None, variance_type=None): method step (line 229) | def step( method add_noise (line 323) | def add_noise( method get_velocity (line 346) | def get_velocity( method __len__ (line 366) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_ddpm_flax.py class DDPMSchedulerState (line 36) | class DDPMSchedulerState: method create (line 45) | def create(cls, common: CommonSchedulerState, init_noise_sigma: jnp.nd... class FlaxDDPMSchedulerOutput (line 50) | class FlaxDDPMSchedulerOutput(FlaxSchedulerOutput): class FlaxDDPMScheduler (line 54) | class FlaxDDPMScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 92) | def has_state(self): method __init__ (line 96) | def __init__( method create_state (line 110) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method scale_model_input (line 125) | def scale_model_input( method set_timesteps (line 139) | def set_timesteps( method _get_variance (line 162) | def _get_variance(self, state: DDPMSchedulerState, t, predicted_varian... method step (line 195) | def step( method add_noise (line 280) | def add_noise( method get_velocity (line 289) | def get_velocity( method __len__ (line 298) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_deis_multistep.py function betas_for_alpha_bar (line 29) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class DEISMultistepScheduler (line 58) | class DEISMultistepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 113) | def __init__( method set_timesteps (line 174) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method convert_model_output (line 197) | def convert_model_output( method deis_first_order_update (line 247) | def deis_first_order_update( method multistep_deis_second_order_update (line 277) | def multistep_deis_second_order_update( method multistep_deis_third_order_update (line 319) | def multistep_deis_third_order_update( method step (line 376) | def step( method scale_model_input (line 444) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 457) | def add_noise( method __len__ (line 480) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_dpmsolver_multistep.py function betas_for_alpha_bar (line 28) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class DPMSolverMultistepScheduler (line 57) | class DPMSolverMultistepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 124) | def __init__( method set_timesteps (line 184) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method convert_model_output (line 207) | def convert_model_output( method dpm_solver_first_order_update (line 278) | def dpm_solver_first_order_update( method multistep_dpm_solver_second_order_update (line 310) | def multistep_dpm_solver_second_order_update( method multistep_dpm_solver_third_order_update (line 369) | def multistep_dpm_solver_third_order_update( method step (line 424) | def step( method scale_model_input (line 492) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 505) | def add_noise( method __len__ (line 528) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_dpmsolver_multistep_flax.py class DPMSolverMultistepSchedulerState (line 35) | class DPMSolverMultistepSchedulerState: method create (line 53) | def create( class FlaxDPMSolverMultistepSchedulerOutput (line 73) | class FlaxDPMSolverMultistepSchedulerOutput(FlaxSchedulerOutput): class FlaxDPMSolverMultistepScheduler (line 77) | class FlaxDPMSolverMultistepScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 147) | def has_state(self): method __init__ (line 151) | def __init__( method create_state (line 170) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method set_timesteps (line 199) | def set_timesteps( method convert_model_output (line 236) | def convert_model_output( method dpm_solver_first_order_update (line 306) | def dpm_solver_first_order_update( method multistep_dpm_solver_second_order_update (line 341) | def multistep_dpm_solver_second_order_update( method multistep_dpm_solver_third_order_update (line 401) | def multistep_dpm_solver_third_order_update( method step (line 457) | def step( method scale_model_input (line 594) | def scale_model_input( method add_noise (line 612) | def add_noise( method __len__ (line 621) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py function betas_for_alpha_bar (line 28) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class DPMSolverSinglestepScheduler (line 57) | class DPMSolverSinglestepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 123) | def __init__( method get_order_list (line 184) | def get_order_list(self, num_inference_steps: int) -> List[int]: method set_timesteps (line 218) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method convert_model_output (line 240) | def convert_model_output( method dpm_solver_first_order_update (line 311) | def dpm_solver_first_order_update( method singlestep_dpm_solver_second_order_update (line 343) | def singlestep_dpm_solver_second_order_update( method singlestep_dpm_solver_third_order_update (line 404) | def singlestep_dpm_solver_third_order_update( method singlestep_dpm_solver_update (line 475) | def singlestep_dpm_solver_update( method step (line 512) | def step( method scale_model_input (line 568) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 581) | def add_noise( method __len__ (line 604) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py class EulerAncestralDiscreteSchedulerOutput (line 32) | class EulerAncestralDiscreteSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999) -> torc... class EulerAncestralDiscreteScheduler (line 79) | class EulerAncestralDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 109) | def __init__( method scale_model_input (line 149) | def scale_model_input( method set_timesteps (line 170) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 193) | def step( method add_noise (line 282) | def add_noise( method __len__ (line 308) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_euler_discrete.py class EulerDiscreteSchedulerOutput (line 32) | class EulerDiscreteSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 50) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class EulerDiscreteScheduler (line 79) | class EulerDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 112) | def __init__( method scale_model_input (line 153) | def scale_model_input( method set_timesteps (line 176) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 209) | def step( method add_noise (line 308) | def add_noise( method __len__ (line 334) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_heun_discrete.py function betas_for_alpha_bar (line 26) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999) -> torc... class HeunDiscreteScheduler (line 55) | class HeunDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 85) | def __init__( method index_for_timestep (line 115) | def index_for_timestep(self, timestep): method scale_model_input (line 123) | def scale_model_input( method set_timesteps (line 142) | def set_timesteps( method state_in_first_order (line 186) | def state_in_first_order(self): method step (line 189) | def step( method add_noise (line 273) | def add_noise( method __len__ (line 298) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_ipndm.py class IPNDMScheduler (line 25) | class IPNDMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 44) | def __init__( method set_timesteps (line 61) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 85) | def step( method scale_model_input (line 135) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method _get_prev_sample (line 148) | def _get_prev_sample(self, sample, timestep_index, prev_timestep_index... method __len__ (line 160) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py function betas_for_alpha_bar (line 27) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999) -> torc... class KDPM2AncestralDiscreteScheduler (line 56) | class KDPM2AncestralDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 87) | def __init__( method index_for_timestep (line 117) | def index_for_timestep(self, timestep): method scale_model_input (line 125) | def scale_model_input( method set_timesteps (line 148) | def set_timesteps( method sigma_to_t (line 211) | def sigma_to_t(self, sigma): method state_in_first_order (line 235) | def state_in_first_order(self): method step (line 238) | def step( method add_noise (line 326) | def add_noise( method __len__ (line 351) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_k_dpm_2_discrete.py function betas_for_alpha_bar (line 26) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999) -> torc... class KDPM2DiscreteScheduler (line 55) | class KDPM2DiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 86) | def __init__( method index_for_timestep (line 116) | def index_for_timestep(self, timestep): method scale_model_input (line 124) | def scale_model_input( method set_timesteps (line 147) | def set_timesteps( method sigma_to_t (line 200) | def sigma_to_t(self, sigma): method state_in_first_order (line 224) | def state_in_first_order(self): method step (line 227) | def step( method add_noise (line 307) | def add_noise( method __len__ (line 332) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_karras_ve.py class KarrasVeOutput (line 28) | class KarrasVeOutput(BaseOutput): class KarrasVeScheduler (line 48) | class KarrasVeScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 83) | def __init__( method scale_model_input (line 100) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 114) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method add_noise_to_input (line 135) | def add_noise_to_input( method step (line 156) | def step( method step_correct (line 194) | def step_correct( method add_noise (line 231) | def add_noise(self, original_samples, noise, timesteps): FILE: src/diffusers/schedulers/scheduling_karras_ve_flax.py class KarrasVeSchedulerState (line 29) | class KarrasVeSchedulerState: method create (line 36) | def create(cls): class FlaxKarrasVeOutput (line 41) | class FlaxKarrasVeOutput(BaseOutput): class FlaxKarrasVeScheduler (line 59) | class FlaxKarrasVeScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 91) | def has_state(self): method __init__ (line 95) | def __init__( method create_state (line 106) | def create_state(self): method set_timesteps (line 109) | def set_timesteps( method add_noise_to_input (line 137) | def add_noise_to_input( method step (line 163) | def step( method step_correct (line 199) | def step_correct( method add_noise (line 236) | def add_noise(self, state: KarrasVeSchedulerState, original_samples, n... FILE: src/diffusers/schedulers/scheduling_lms_discrete.py class LMSDiscreteSchedulerOutput (line 30) | class LMSDiscreteSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 48) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class LMSDiscreteScheduler (line 77) | class LMSDiscreteScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 107) | def __init__( method scale_model_input (line 148) | def scale_model_input( method get_lms_coefficient (line 169) | def get_lms_coefficient(self, order, t, current_order): method set_timesteps (line 191) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 217) | def step( method add_noise (line 287) | def add_noise( method __len__ (line 312) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_lms_discrete_flax.py class LMSDiscreteSchedulerState (line 33) | class LMSDiscreteSchedulerState: method create (line 46) | def create( class FlaxLMSSchedulerOutput (line 53) | class FlaxLMSSchedulerOutput(FlaxSchedulerOutput): class FlaxLMSDiscreteScheduler (line 57) | class FlaxLMSDiscreteScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 90) | def has_state(self): method __init__ (line 94) | def __init__( method create_state (line 106) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method scale_model_input (line 123) | def scale_model_input(self, state: LMSDiscreteSchedulerState, sample: ... method get_lms_coefficient (line 145) | def get_lms_coefficient(self, state: LMSDiscreteSchedulerState, order,... method set_timesteps (line 167) | def set_timesteps( method step (line 203) | def step( method add_noise (line 268) | def add_noise( method __len__ (line 282) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_pndm.py function betas_for_alpha_bar (line 28) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class PNDMScheduler (line 57) | class PNDMScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 99) | def __init__( method set_timesteps (line 152) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 192) | def step( method step_prk (line 223) | def step_prk( method step_plms (line 278) | def step_plms( method scale_model_input (line 345) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method _get_prev_sample (line 358) | def _get_prev_sample(self, sample, timestep, prev_timestep, model_outp... method add_noise (line 401) | def add_noise( method __len__ (line 424) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_pndm_flax.py class PNDMSchedulerState (line 35) | class PNDMSchedulerState: method create (line 53) | def create( class FlaxPNDMSchedulerOutput (line 69) | class FlaxPNDMSchedulerOutput(FlaxSchedulerOutput): class FlaxPNDMScheduler (line 73) | class FlaxPNDMScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 119) | def has_state(self): method __init__ (line 123) | def __init__( method create_state (line 143) | def create_state(self, common: Optional[CommonSchedulerState] = None) ... method set_timesteps (line 167) | def set_timesteps(self, state: PNDMSchedulerState, num_inference_steps... method scale_model_input (line 222) | def scale_model_input( method step (line 239) | def step( method step_prk (line 294) | def step_prk( method step_plms (line 362) | def step_plms( method _get_prev_sample (line 456) | def _get_prev_sample(self, state: PNDMSchedulerState, sample, timestep... method add_noise (line 501) | def add_noise( method __len__ (line 510) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_repaint.py class RePaintSchedulerOutput (line 28) | class RePaintSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 46) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class RePaintScheduler (line 75) | class RePaintScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 109) | def __init__( method scale_model_input (line 153) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 167) | def set_timesteps( method _get_variance (line 197) | def _get_variance(self, t): method step (line 216) | def step( method undo_step (line 303) | def undo_step(self, sample, timestep, generator=None): method add_noise (line 320) | def add_noise( method __len__ (line 328) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_sde_ve.py class SdeVeOutput (line 29) | class SdeVeOutput(BaseOutput): class ScoreSdeVeScheduler (line 45) | class ScoreSdeVeScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 72) | def __init__( method scale_model_input (line 89) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 103) | def set_timesteps( method set_sigmas (line 119) | def set_sigmas( method get_adjacent_sigma (line 146) | def get_adjacent_sigma(self, timesteps, t): method step_pred (line 153) | def step_pred( method step_correct (line 216) | def step_correct( method add_noise (line 267) | def add_noise( method __len__ (line 280) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_sde_ve_flax.py class ScoreSdeVeSchedulerState (line 29) | class ScoreSdeVeSchedulerState: method create (line 36) | def create(cls): class FlaxSdeVeOutput (line 41) | class FlaxSdeVeOutput(FlaxSchedulerOutput): class FlaxScoreSdeVeScheduler (line 59) | class FlaxScoreSdeVeScheduler(FlaxSchedulerMixin, ConfigMixin): method has_state (line 84) | def has_state(self): method __init__ (line 88) | def __init__( method create_state (line 99) | def create_state(self): method set_timesteps (line 109) | def set_timesteps( method set_sigmas (line 127) | def set_sigmas( method get_adjacent_sigma (line 160) | def get_adjacent_sigma(self, state, timesteps, t): method step_pred (line 163) | def step_pred( method step_correct (line 223) | def step_correct( method __len__ (line 275) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_sde_vp.py class ScoreSdeVpScheduler (line 27) | class ScoreSdeVpScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 45) | def __init__(self, num_train_timesteps=2000, beta_min=0.1, beta_max=20... method set_timesteps (line 50) | def set_timesteps(self, num_inference_steps, device: Union[str, torch.... method step_pred (line 53) | def step_pred(self, score, x, t, generator=None): method __len__ (line 89) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_unclip.py class UnCLIPSchedulerOutput (line 29) | class UnCLIPSchedulerOutput(BaseOutput): function betas_for_alpha_bar (line 47) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class UnCLIPScheduler (line 76) | class UnCLIPScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 103) | def __init__( method scale_model_input (line 130) | def scale_model_input(self, sample: torch.FloatTensor, timestep: Optio... method set_timesteps (line 144) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method _get_variance (line 161) | def _get_variance(self, t, prev_timestep=None, predicted_variance=None... method step (line 197) | def step( FILE: src/diffusers/schedulers/scheduling_unipc_multistep.py function betas_for_alpha_bar (line 28) | def betas_for_alpha_bar(num_diffusion_timesteps, max_beta=0.999): class UniPCMultistepScheduler (line 57) | class UniPCMultistepScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 126) | def __init__( method set_timesteps (line 187) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method convert_model_output (line 213) | def convert_model_output( method multistep_uni_p_bh_update (line 275) | def multistep_uni_p_bh_update( method multistep_uni_c_bh_update (line 380) | def multistep_uni_c_bh_update( method step (line 486) | def step( method scale_model_input (line 570) | def scale_model_input(self, sample: torch.FloatTensor, *args, **kwargs... method add_noise (line 583) | def add_noise( method __len__ (line 606) | def __len__(self): FILE: src/diffusers/schedulers/scheduling_utils.py class KarrasDiffusionSchedulers (line 32) | class KarrasDiffusionSchedulers(Enum): class SchedulerOutput (line 49) | class SchedulerOutput(BaseOutput): class SchedulerMixin (line 62) | class SchedulerMixin: method from_pretrained (line 77) | def from_pretrained( method save_pretrained (line 147) | def save_pretrained(self, save_directory: Union[str, os.PathLike], pus... method compatibles (line 159) | def compatibles(self): method _get_compatibles (line 169) | def _get_compatibles(cls): FILE: src/diffusers/schedulers/scheduling_utils_flax.py class FlaxKarrasDiffusionSchedulers (line 34) | class FlaxKarrasDiffusionSchedulers(Enum): class FlaxSchedulerOutput (line 43) | class FlaxSchedulerOutput(BaseOutput): class FlaxSchedulerMixin (line 56) | class FlaxSchedulerMixin: method from_pretrained (line 72) | def from_pretrained( method save_pretrained (line 151) | def save_pretrained(self, save_directory: Union[str, os.PathLike], pus... method compatibles (line 163) | def compatibles(self): method _get_compatibles (line 173) | def _get_compatibles(cls): function broadcast_to_shape_from_left (line 182) | def broadcast_to_shape_from_left(x: jnp.ndarray, shape: Tuple[int]) -> j... function betas_for_alpha_bar (line 187) | def betas_for_alpha_bar(num_diffusion_timesteps: int, max_beta=0.999, dt... class CommonSchedulerState (line 217) | class CommonSchedulerState: method create (line 223) | def create(cls, scheduler): function get_sqrt_alpha_prod (line 257) | def get_sqrt_alpha_prod( function add_noise_common (line 273) | def add_noise_common( function get_velocity_common (line 281) | def get_velocity_common(state: CommonSchedulerState, sample: jnp.ndarray... FILE: src/diffusers/schedulers/scheduling_vq_diffusion.py class VQDiffusionSchedulerOutput (line 28) | class VQDiffusionSchedulerOutput(BaseOutput): function index_to_log_onehot (line 41) | def index_to_log_onehot(x: torch.LongTensor, num_classes: int) -> torch.... function gumbel_noised (line 62) | def gumbel_noised(logits: torch.FloatTensor, generator: Optional[torch.G... function alpha_schedules (line 72) | def alpha_schedules(num_diffusion_timesteps: int, alpha_cum_start=0.9999... function gamma_schedules (line 88) | def gamma_schedules(num_diffusion_timesteps: int, gamma_cum_start=0.0000... class VQDiffusionScheduler (line 106) | class VQDiffusionScheduler(SchedulerMixin, ConfigMixin): method __init__ (line 144) | def __init__( method set_timesteps (line 190) | def set_timesteps(self, num_inference_steps: int, device: Union[str, t... method step (line 212) | def step( method q_posterior (line 260) | def q_posterior(self, log_p_x_0, x_t, t): method log_Q_t_transitioning_to_known_class (line 379) | def log_Q_t_transitioning_to_known_class( method apply_cumulative_transitions (line 484) | def apply_cumulative_transitions(self, q, t): FILE: src/diffusers/training_utils.py function enable_full_determinism (line 12) | def enable_full_determinism(seed: int): function set_seed (line 32) | def set_seed(seed: int): class EMAModel (line 46) | class EMAModel: method __init__ (line 51) | def __init__( method from_pretrained (line 133) | def from_pretrained(cls, path, model_cls) -> "EMAModel": method save_pretrained (line 142) | def save_pretrained(self, path): method get_decay (line 157) | def get_decay(self, optimization_step: int) -> float: method step (line 177) | def step(self, parameters: Iterable[torch.nn.Parameter]): method copy_to (line 208) | def copy_to(self, parameters: Iterable[torch.nn.Parameter]) -> None: method to (line 221) | def to(self, device=None, dtype=None) -> None: method state_dict (line 233) | def state_dict(self) -> dict: method store (line 252) | def store(self, parameters: Iterable[torch.nn.Parameter]) -> None: method restore (line 261) | def restore(self, parameters: Iterable[torch.nn.Parameter]) -> None: method load_state_dict (line 279) | def load_state_dict(self, state_dict: dict) -> None: FILE: src/diffusers/utils/__init__.py function check_min_version (line 95) | def check_min_version(min_version): FILE: src/diffusers/utils/accelerate_utils.py function apply_forward_hook (line 27) | def apply_forward_hook(method): FILE: src/diffusers/utils/deprecation_utils.py function deprecate (line 8) | def deprecate(*args, take_from: Optional[Union[Dict, Any]] = None, stand... FILE: src/diffusers/utils/doc_utils.py function replace_example_docstring (line 20) | def replace_example_docstring(example_docstring): FILE: src/diffusers/utils/dummy_flax_and_transformers_objects.py class FlaxStableDiffusionImg2ImgPipeline (line 5) | class FlaxStableDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class FlaxStableDiffusionInpaintPipeline (line 20) | class FlaxStableDiffusionInpaintPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class FlaxStableDiffusionPipeline (line 35) | class FlaxStableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): FILE: src/diffusers/utils/dummy_flax_objects.py class FlaxModelMixin (line 5) | class FlaxModelMixin(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class FlaxUNet2DConditionModel (line 20) | class FlaxUNet2DConditionModel(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class FlaxAutoencoderKL (line 35) | class FlaxAutoencoderKL(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class FlaxDiffusionPipeline (line 50) | class FlaxDiffusionPipeline(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class FlaxDDIMScheduler (line 65) | class FlaxDDIMScheduler(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class FlaxDDPMScheduler (line 80) | class FlaxDDPMScheduler(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): class FlaxDPMSolverMultistepScheduler (line 95) | class FlaxDPMSolverMultistepScheduler(metaclass=DummyObject): method __init__ (line 98) | def __init__(self, *args, **kwargs): method from_config (line 102) | def from_config(cls, *args, **kwargs): method from_pretrained (line 106) | def from_pretrained(cls, *args, **kwargs): class FlaxKarrasVeScheduler (line 110) | class FlaxKarrasVeScheduler(metaclass=DummyObject): method __init__ (line 113) | def __init__(self, *args, **kwargs): method from_config (line 117) | def from_config(cls, *args, **kwargs): method from_pretrained (line 121) | def from_pretrained(cls, *args, **kwargs): class FlaxLMSDiscreteScheduler (line 125) | class FlaxLMSDiscreteScheduler(metaclass=DummyObject): method __init__ (line 128) | def __init__(self, *args, **kwargs): method from_config (line 132) | def from_config(cls, *args, **kwargs): method from_pretrained (line 136) | def from_pretrained(cls, *args, **kwargs): class FlaxPNDMScheduler (line 140) | class FlaxPNDMScheduler(metaclass=DummyObject): method __init__ (line 143) | def __init__(self, *args, **kwargs): method from_config (line 147) | def from_config(cls, *args, **kwargs): method from_pretrained (line 151) | def from_pretrained(cls, *args, **kwargs): class FlaxSchedulerMixin (line 155) | class FlaxSchedulerMixin(metaclass=DummyObject): method __init__ (line 158) | def __init__(self, *args, **kwargs): method from_config (line 162) | def from_config(cls, *args, **kwargs): method from_pretrained (line 166) | def from_pretrained(cls, *args, **kwargs): class FlaxScoreSdeVeScheduler (line 170) | class FlaxScoreSdeVeScheduler(metaclass=DummyObject): method __init__ (line 173) | def __init__(self, *args, **kwargs): method from_config (line 177) | def from_config(cls, *args, **kwargs): method from_pretrained (line 181) | def from_pretrained(cls, *args, **kwargs): FILE: src/diffusers/utils/dummy_onnx_objects.py class OnnxRuntimeModel (line 5) | class OnnxRuntimeModel(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: src/diffusers/utils/dummy_pt_objects.py class AutoencoderKL (line 5) | class AutoencoderKL(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class ControlNetModel (line 20) | class ControlNetModel(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class ModelMixin (line 35) | class ModelMixin(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class PriorTransformer (line 50) | class PriorTransformer(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class Transformer2DModel (line 65) | class Transformer2DModel(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class UNet1DModel (line 80) | class UNet1DModel(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): class UNet2DConditionModel (line 95) | class UNet2DConditionModel(metaclass=DummyObject): method __init__ (line 98) | def __init__(self, *args, **kwargs): method from_config (line 102) | def from_config(cls, *args, **kwargs): method from_pretrained (line 106) | def from_pretrained(cls, *args, **kwargs): class UNet2DModel (line 110) | class UNet2DModel(metaclass=DummyObject): method __init__ (line 113) | def __init__(self, *args, **kwargs): method from_config (line 117) | def from_config(cls, *args, **kwargs): method from_pretrained (line 121) | def from_pretrained(cls, *args, **kwargs): class VQModel (line 125) | class VQModel(metaclass=DummyObject): method __init__ (line 128) | def __init__(self, *args, **kwargs): method from_config (line 132) | def from_config(cls, *args, **kwargs): method from_pretrained (line 136) | def from_pretrained(cls, *args, **kwargs): function get_constant_schedule (line 140) | def get_constant_schedule(*args, **kwargs): function get_constant_schedule_with_warmup (line 144) | def get_constant_schedule_with_warmup(*args, **kwargs): function get_cosine_schedule_with_warmup (line 148) | def get_cosine_schedule_with_warmup(*args, **kwargs): function get_cosine_with_hard_restarts_schedule_with_warmup (line 152) | def get_cosine_with_hard_restarts_schedule_with_warmup(*args, **kwargs): function get_linear_schedule_with_warmup (line 156) | def get_linear_schedule_with_warmup(*args, **kwargs): function get_polynomial_decay_schedule_with_warmup (line 160) | def get_polynomial_decay_schedule_with_warmup(*args, **kwargs): function get_scheduler (line 164) | def get_scheduler(*args, **kwargs): class AudioPipelineOutput (line 168) | class AudioPipelineOutput(metaclass=DummyObject): method __init__ (line 171) | def __init__(self, *args, **kwargs): method from_config (line 175) | def from_config(cls, *args, **kwargs): method from_pretrained (line 179) | def from_pretrained(cls, *args, **kwargs): class DanceDiffusionPipeline (line 183) | class DanceDiffusionPipeline(metaclass=DummyObject): method __init__ (line 186) | def __init__(self, *args, **kwargs): method from_config (line 190) | def from_config(cls, *args, **kwargs): method from_pretrained (line 194) | def from_pretrained(cls, *args, **kwargs): class DDIMPipeline (line 198) | class DDIMPipeline(metaclass=DummyObject): method __init__ (line 201) | def __init__(self, *args, **kwargs): method from_config (line 205) | def from_config(cls, *args, **kwargs): method from_pretrained (line 209) | def from_pretrained(cls, *args, **kwargs): class DDPMPipeline (line 213) | class DDPMPipeline(metaclass=DummyObject): method __init__ (line 216) | def __init__(self, *args, **kwargs): method from_config (line 220) | def from_config(cls, *args, **kwargs): method from_pretrained (line 224) | def from_pretrained(cls, *args, **kwargs): class DiffusionPipeline (line 228) | class DiffusionPipeline(metaclass=DummyObject): method __init__ (line 231) | def __init__(self, *args, **kwargs): method from_config (line 235) | def from_config(cls, *args, **kwargs): method from_pretrained (line 239) | def from_pretrained(cls, *args, **kwargs): class DiTPipeline (line 243) | class DiTPipeline(metaclass=DummyObject): method __init__ (line 246) | def __init__(self, *args, **kwargs): method from_config (line 250) | def from_config(cls, *args, **kwargs): method from_pretrained (line 254) | def from_pretrained(cls, *args, **kwargs): class ImagePipelineOutput (line 258) | class ImagePipelineOutput(metaclass=DummyObject): method __init__ (line 261) | def __init__(self, *args, **kwargs): method from_config (line 265) | def from_config(cls, *args, **kwargs): method from_pretrained (line 269) | def from_pretrained(cls, *args, **kwargs): class KarrasVePipeline (line 273) | class KarrasVePipeline(metaclass=DummyObject): method __init__ (line 276) | def __init__(self, *args, **kwargs): method from_config (line 280) | def from_config(cls, *args, **kwargs): method from_pretrained (line 284) | def from_pretrained(cls, *args, **kwargs): class LDMPipeline (line 288) | class LDMPipeline(metaclass=DummyObject): method __init__ (line 291) | def __init__(self, *args, **kwargs): method from_config (line 295) | def from_config(cls, *args, **kwargs): method from_pretrained (line 299) | def from_pretrained(cls, *args, **kwargs): class LDMSuperResolutionPipeline (line 303) | class LDMSuperResolutionPipeline(metaclass=DummyObject): method __init__ (line 306) | def __init__(self, *args, **kwargs): method from_config (line 310) | def from_config(cls, *args, **kwargs): method from_pretrained (line 314) | def from_pretrained(cls, *args, **kwargs): class PNDMPipeline (line 318) | class PNDMPipeline(metaclass=DummyObject): method __init__ (line 321) | def __init__(self, *args, **kwargs): method from_config (line 325) | def from_config(cls, *args, **kwargs): method from_pretrained (line 329) | def from_pretrained(cls, *args, **kwargs): class RePaintPipeline (line 333) | class RePaintPipeline(metaclass=DummyObject): method __init__ (line 336) | def __init__(self, *args, **kwargs): method from_config (line 340) | def from_config(cls, *args, **kwargs): method from_pretrained (line 344) | def from_pretrained(cls, *args, **kwargs): class ScoreSdeVePipeline (line 348) | class ScoreSdeVePipeline(metaclass=DummyObject): method __init__ (line 351) | def __init__(self, *args, **kwargs): method from_config (line 355) | def from_config(cls, *args, **kwargs): method from_pretrained (line 359) | def from_pretrained(cls, *args, **kwargs): class DDIMInverseScheduler (line 363) | class DDIMInverseScheduler(metaclass=DummyObject): method __init__ (line 366) | def __init__(self, *args, **kwargs): method from_config (line 370) | def from_config(cls, *args, **kwargs): method from_pretrained (line 374) | def from_pretrained(cls, *args, **kwargs): class DDIMScheduler (line 378) | class DDIMScheduler(metaclass=DummyObject): method __init__ (line 381) | def __init__(self, *args, **kwargs): method from_config (line 385) | def from_config(cls, *args, **kwargs): method from_pretrained (line 389) | def from_pretrained(cls, *args, **kwargs): class DDPMScheduler (line 393) | class DDPMScheduler(metaclass=DummyObject): method __init__ (line 396) | def __init__(self, *args, **kwargs): method from_config (line 400) | def from_config(cls, *args, **kwargs): method from_pretrained (line 404) | def from_pretrained(cls, *args, **kwargs): class DEISMultistepScheduler (line 408) | class DEISMultistepScheduler(metaclass=DummyObject): method __init__ (line 411) | def __init__(self, *args, **kwargs): method from_config (line 415) | def from_config(cls, *args, **kwargs): method from_pretrained (line 419) | def from_pretrained(cls, *args, **kwargs): class DPMSolverMultistepScheduler (line 423) | class DPMSolverMultistepScheduler(metaclass=DummyObject): method __init__ (line 426) | def __init__(self, *args, **kwargs): method from_config (line 430) | def from_config(cls, *args, **kwargs): method from_pretrained (line 434) | def from_pretrained(cls, *args, **kwargs): class DPMSolverSinglestepScheduler (line 438) | class DPMSolverSinglestepScheduler(metaclass=DummyObject): method __init__ (line 441) | def __init__(self, *args, **kwargs): method from_config (line 445) | def from_config(cls, *args, **kwargs): method from_pretrained (line 449) | def from_pretrained(cls, *args, **kwargs): class EulerAncestralDiscreteScheduler (line 453) | class EulerAncestralDiscreteScheduler(metaclass=DummyObject): method __init__ (line 456) | def __init__(self, *args, **kwargs): method from_config (line 460) | def from_config(cls, *args, **kwargs): method from_pretrained (line 464) | def from_pretrained(cls, *args, **kwargs): class EulerDiscreteScheduler (line 468) | class EulerDiscreteScheduler(metaclass=DummyObject): method __init__ (line 471) | def __init__(self, *args, **kwargs): method from_config (line 475) | def from_config(cls, *args, **kwargs): method from_pretrained (line 479) | def from_pretrained(cls, *args, **kwargs): class HeunDiscreteScheduler (line 483) | class HeunDiscreteScheduler(metaclass=DummyObject): method __init__ (line 486) | def __init__(self, *args, **kwargs): method from_config (line 490) | def from_config(cls, *args, **kwargs): method from_pretrained (line 494) | def from_pretrained(cls, *args, **kwargs): class IPNDMScheduler (line 498) | class IPNDMScheduler(metaclass=DummyObject): method __init__ (line 501) | def __init__(self, *args, **kwargs): method from_config (line 505) | def from_config(cls, *args, **kwargs): method from_pretrained (line 509) | def from_pretrained(cls, *args, **kwargs): class KarrasVeScheduler (line 513) | class KarrasVeScheduler(metaclass=DummyObject): method __init__ (line 516) | def __init__(self, *args, **kwargs): method from_config (line 520) | def from_config(cls, *args, **kwargs): method from_pretrained (line 524) | def from_pretrained(cls, *args, **kwargs): class KDPM2AncestralDiscreteScheduler (line 528) | class KDPM2AncestralDiscreteScheduler(metaclass=DummyObject): method __init__ (line 531) | def __init__(self, *args, **kwargs): method from_config (line 535) | def from_config(cls, *args, **kwargs): method from_pretrained (line 539) | def from_pretrained(cls, *args, **kwargs): class KDPM2DiscreteScheduler (line 543) | class KDPM2DiscreteScheduler(metaclass=DummyObject): method __init__ (line 546) | def __init__(self, *args, **kwargs): method from_config (line 550) | def from_config(cls, *args, **kwargs): method from_pretrained (line 554) | def from_pretrained(cls, *args, **kwargs): class PNDMScheduler (line 558) | class PNDMScheduler(metaclass=DummyObject): method __init__ (line 561) | def __init__(self, *args, **kwargs): method from_config (line 565) | def from_config(cls, *args, **kwargs): method from_pretrained (line 569) | def from_pretrained(cls, *args, **kwargs): class RePaintScheduler (line 573) | class RePaintScheduler(metaclass=DummyObject): method __init__ (line 576) | def __init__(self, *args, **kwargs): method from_config (line 580) | def from_config(cls, *args, **kwargs): method from_pretrained (line 584) | def from_pretrained(cls, *args, **kwargs): class SchedulerMixin (line 588) | class SchedulerMixin(metaclass=DummyObject): method __init__ (line 591) | def __init__(self, *args, **kwargs): method from_config (line 595) | def from_config(cls, *args, **kwargs): method from_pretrained (line 599) | def from_pretrained(cls, *args, **kwargs): class ScoreSdeVeScheduler (line 603) | class ScoreSdeVeScheduler(metaclass=DummyObject): method __init__ (line 606) | def __init__(self, *args, **kwargs): method from_config (line 610) | def from_config(cls, *args, **kwargs): method from_pretrained (line 614) | def from_pretrained(cls, *args, **kwargs): class UnCLIPScheduler (line 618) | class UnCLIPScheduler(metaclass=DummyObject): method __init__ (line 621) | def __init__(self, *args, **kwargs): method from_config (line 625) | def from_config(cls, *args, **kwargs): method from_pretrained (line 629) | def from_pretrained(cls, *args, **kwargs): class UniPCMultistepScheduler (line 633) | class UniPCMultistepScheduler(metaclass=DummyObject): method __init__ (line 636) | def __init__(self, *args, **kwargs): method from_config (line 640) | def from_config(cls, *args, **kwargs): method from_pretrained (line 644) | def from_pretrained(cls, *args, **kwargs): class VQDiffusionScheduler (line 648) | class VQDiffusionScheduler(metaclass=DummyObject): method __init__ (line 651) | def __init__(self, *args, **kwargs): method from_config (line 655) | def from_config(cls, *args, **kwargs): method from_pretrained (line 659) | def from_pretrained(cls, *args, **kwargs): class EMAModel (line 663) | class EMAModel(metaclass=DummyObject): method __init__ (line 666) | def __init__(self, *args, **kwargs): method from_config (line 670) | def from_config(cls, *args, **kwargs): method from_pretrained (line 674) | def from_pretrained(cls, *args, **kwargs): FILE: src/diffusers/utils/dummy_torch_and_librosa_objects.py class AudioDiffusionPipeline (line 5) | class AudioDiffusionPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class Mel (line 20) | class Mel(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): FILE: src/diffusers/utils/dummy_torch_and_scipy_objects.py class LMSDiscreteScheduler (line 5) | class LMSDiscreteScheduler(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: src/diffusers/utils/dummy_torch_and_transformers_and_k_diffusion_objects.py class StableDiffusionKDiffusionPipeline (line 5) | class StableDiffusionKDiffusionPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): FILE: src/diffusers/utils/dummy_torch_and_transformers_and_onnx_objects.py class OnnxStableDiffusionImg2ImgPipeline (line 5) | class OnnxStableDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionInpaintPipeline (line 20) | class OnnxStableDiffusionInpaintPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionInpaintPipelineLegacy (line 35) | class OnnxStableDiffusionInpaintPipelineLegacy(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class OnnxStableDiffusionPipeline (line 50) | class OnnxStableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionOnnxPipeline (line 65) | class StableDiffusionOnnxPipeline(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): FILE: src/diffusers/utils/dummy_torch_and_transformers_objects.py class AltDiffusionImg2ImgPipeline (line 5) | class AltDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 8) | def __init__(self, *args, **kwargs): method from_config (line 12) | def from_config(cls, *args, **kwargs): method from_pretrained (line 16) | def from_pretrained(cls, *args, **kwargs): class AltDiffusionPipeline (line 20) | class AltDiffusionPipeline(metaclass=DummyObject): method __init__ (line 23) | def __init__(self, *args, **kwargs): method from_config (line 27) | def from_config(cls, *args, **kwargs): method from_pretrained (line 31) | def from_pretrained(cls, *args, **kwargs): class CycleDiffusionPipeline (line 35) | class CycleDiffusionPipeline(metaclass=DummyObject): method __init__ (line 38) | def __init__(self, *args, **kwargs): method from_config (line 42) | def from_config(cls, *args, **kwargs): method from_pretrained (line 46) | def from_pretrained(cls, *args, **kwargs): class LDMTextToImagePipeline (line 50) | class LDMTextToImagePipeline(metaclass=DummyObject): method __init__ (line 53) | def __init__(self, *args, **kwargs): method from_config (line 57) | def from_config(cls, *args, **kwargs): method from_pretrained (line 61) | def from_pretrained(cls, *args, **kwargs): class PaintByExamplePipeline (line 65) | class PaintByExamplePipeline(metaclass=DummyObject): method __init__ (line 68) | def __init__(self, *args, **kwargs): method from_config (line 72) | def from_config(cls, *args, **kwargs): method from_pretrained (line 76) | def from_pretrained(cls, *args, **kwargs): class SemanticStableDiffusionPipeline (line 80) | class SemanticStableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 83) | def __init__(self, *args, **kwargs): method from_config (line 87) | def from_config(cls, *args, **kwargs): method from_pretrained (line 91) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionAttendAndExcitePipeline (line 95) | class StableDiffusionAttendAndExcitePipeline(metaclass=DummyObject): method __init__ (line 98) | def __init__(self, *args, **kwargs): method from_config (line 102) | def from_config(cls, *args, **kwargs): method from_pretrained (line 106) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionControlNetPipeline (line 110) | class StableDiffusionControlNetPipeline(metaclass=DummyObject): method __init__ (line 113) | def __init__(self, *args, **kwargs): method from_config (line 117) | def from_config(cls, *args, **kwargs): method from_pretrained (line 121) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionDepth2ImgPipeline (line 125) | class StableDiffusionDepth2ImgPipeline(metaclass=DummyObject): method __init__ (line 128) | def __init__(self, *args, **kwargs): method from_config (line 132) | def from_config(cls, *args, **kwargs): method from_pretrained (line 136) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionImageVariationPipeline (line 140) | class StableDiffusionImageVariationPipeline(metaclass=DummyObject): method __init__ (line 143) | def __init__(self, *args, **kwargs): method from_config (line 147) | def from_config(cls, *args, **kwargs): method from_pretrained (line 151) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionImg2ImgPipeline (line 155) | class StableDiffusionImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 158) | def __init__(self, *args, **kwargs): method from_config (line 162) | def from_config(cls, *args, **kwargs): method from_pretrained (line 166) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionInpaintPipeline (line 170) | class StableDiffusionInpaintPipeline(metaclass=DummyObject): method __init__ (line 173) | def __init__(self, *args, **kwargs): method from_config (line 177) | def from_config(cls, *args, **kwargs): method from_pretrained (line 181) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionInpaintPipelineLegacy (line 185) | class StableDiffusionInpaintPipelineLegacy(metaclass=DummyObject): method __init__ (line 188) | def __init__(self, *args, **kwargs): method from_config (line 192) | def from_config(cls, *args, **kwargs): method from_pretrained (line 196) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionInstructPix2PixPipeline (line 200) | class StableDiffusionInstructPix2PixPipeline(metaclass=DummyObject): method __init__ (line 203) | def __init__(self, *args, **kwargs): method from_config (line 207) | def from_config(cls, *args, **kwargs): method from_pretrained (line 211) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionLatentUpscalePipeline (line 215) | class StableDiffusionLatentUpscalePipeline(metaclass=DummyObject): method __init__ (line 218) | def __init__(self, *args, **kwargs): method from_config (line 222) | def from_config(cls, *args, **kwargs): method from_pretrained (line 226) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPanoramaPipeline (line 230) | class StableDiffusionPanoramaPipeline(metaclass=DummyObject): method __init__ (line 233) | def __init__(self, *args, **kwargs): method from_config (line 237) | def from_config(cls, *args, **kwargs): method from_pretrained (line 241) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPipeline (line 245) | class StableDiffusionPipeline(metaclass=DummyObject): method __init__ (line 248) | def __init__(self, *args, **kwargs): method from_config (line 252) | def from_config(cls, *args, **kwargs): method from_pretrained (line 256) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPipelineSafe (line 260) | class StableDiffusionPipelineSafe(metaclass=DummyObject): method __init__ (line 263) | def __init__(self, *args, **kwargs): method from_config (line 267) | def from_config(cls, *args, **kwargs): method from_pretrained (line 271) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionPix2PixZeroPipeline (line 275) | class StableDiffusionPix2PixZeroPipeline(metaclass=DummyObject): method __init__ (line 278) | def __init__(self, *args, **kwargs): method from_config (line 282) | def from_config(cls, *args, **kwargs): method from_pretrained (line 286) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionSAGPipeline (line 290) | class StableDiffusionSAGPipeline(metaclass=DummyObject): method __init__ (line 293) | def __init__(self, *args, **kwargs): method from_config (line 297) | def from_config(cls, *args, **kwargs): method from_pretrained (line 301) | def from_pretrained(cls, *args, **kwargs): class StableDiffusionUpscalePipeline (line 305) | class StableDiffusionUpscalePipeline(metaclass=DummyObject): method __init__ (line 308) | def __init__(self, *args, **kwargs): method from_config (line 312) | def from_config(cls, *args, **kwargs): method from_pretrained (line 316) | def from_pretrained(cls, *args, **kwargs): class StableUnCLIPImg2ImgPipeline (line 320) | class StableUnCLIPImg2ImgPipeline(metaclass=DummyObject): method __init__ (line 323) | def __init__(self, *args, **kwargs): method from_config (line 327) | def from_config(cls, *args, **kwargs): method from_pretrained (line 331) | def from_pretrained(cls, *args, **kwargs): class StableUnCLIPPipeline (line 335) | class StableUnCLIPPipeline(metaclass=DummyObject): method __init__ (line 338) | def __init__(self, *args, **kwargs): method from_config (line 342) | def from_config(cls, *args, **kwargs): method from_pretrained (line 346) | def from_pretrained(cls, *args, **kwargs): class UnCLIPImageVariationPipeline (line 350) | class UnCLIPImageVariationPipeline(metaclass=DummyObject): method __init__ (line 353) | def __init__(self, *args, **kwargs): method from_config (line 357) | def from_config(cls, *args, **kwargs): method from_pretrained (line 361) | def from_pretrained(cls, *args, **kwargs): class UnCLIPPipeline (line 365) | class UnCLIPPipeline(metaclass=DummyObject): method __init__ (line 368) | def __init__(self, *args, **kwargs): method from_config (line 372) | def from_config(cls, *args, **kwargs): method from_pretrained (line 376) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionDualGuidedPipeline (line 380) | class VersatileDiffusionDualGuidedPipeline(metaclass=DummyObject): method __init__ (line 383) | def __init__(self, *args, **kwargs): method from_config (line 387) | def from_config(cls, *args, **kwargs): method from_pretrained (line 391) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionImageVariationPipeline (line 395) | class VersatileDiffusionImageVariationPipeline(metaclass=DummyObject): method __init__ (line 398) | def __init__(self, *args, **kwargs): method from_config (line 402) | def from_config(cls, *args, **kwargs): method from_pretrained (line 406) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionPipeline (line 410) | class VersatileDiffusionPipeline(metaclass=DummyObject): method __init__ (line 413) | def __init__(self, *args, **kwargs): method from_config (line 417) | def from_config(cls, *args, **kwargs): method from_pretrained (line 421) | def from_pretrained(cls, *args, **kwargs): class VersatileDiffusionTextToImagePipeline (line 425) | class VersatileDiffusionTextToImagePipeline(metaclass=DummyObject): method __init__ (line 428) | def __init__(self, *args, **kwargs): method from_config (line 432) | def from_config(cls, *args, **kwargs): method from_pretrained (line 436) | def from_pretrained(cls, *args, **kwargs): class VQDiffusionPipeline (line 440) | class VQDiffusionPipeline(metaclass=DummyObject): method __init__ (line 443) | def __init__(self, *args, **kwargs): method from_config (line 447) | def from_config(cls, *args, **kwargs): method from_pretrained (line 451) | def from_pretrained(cls, *args, **kwargs): FILE: src/diffusers/utils/dynamic_modules_utils.py function get_diffusers_versions (line 43) | def get_diffusers_versions(): function init_hf_modules (line 49) | def init_hf_modules(): function create_dynamic_module (line 64) | def create_dynamic_module(name: Union[str, os.PathLike]): function get_relative_imports (line 79) | def get_relative_imports(module_file): function get_relative_import_files (line 97) | def get_relative_import_files(module_file): function check_imports (line 126) | def check_imports(filename): function get_class_in_module (line 158) | def get_class_in_module(class_name, module_path): function find_pipeline_class (line 170) | def find_pipeline_class(loaded_module): function get_cached_module_file (line 197) | def get_cached_module_file( function get_class_from_dynamic_module (line 369) | def get_class_from_dynamic_module( FILE: src/diffusers/utils/hub_utils.py function http_user_agent (line 52) | def http_user_agent(user_agent: Union[Dict, str, None] = None) -> str: function get_full_repo_name (line 76) | def get_full_repo_name(model_id: str, organization: Optional[str] = None... function create_model_card (line 86) | def create_model_card(args, model_name): function move_cache (line 145) | def move_cache(old_cache_dir: Optional[str] = None, new_cache_dir: Optio... FILE: src/diffusers/utils/import_utils.py function is_torch_available (line 235) | def is_torch_available(): function is_safetensors_available (line 239) | def is_safetensors_available(): function is_tf_available (line 243) | def is_tf_available(): function is_flax_available (line 247) | def is_flax_available(): function is_transformers_available (line 251) | def is_transformers_available(): function is_inflect_available (line 255) | def is_inflect_available(): function is_unidecode_available (line 259) | def is_unidecode_available(): function is_onnx_available (line 263) | def is_onnx_available(): function is_scipy_available (line 267) | def is_scipy_available(): function is_librosa_available (line 271) | def is_librosa_available(): function is_xformers_available (line 275) | def is_xformers_available(): function is_accelerate_available (line 279) | def is_accelerate_available(): function is_k_diffusion_available (line 283) | def is_k_diffusion_available(): function is_wandb_available (line 287) | def is_wandb_available(): function is_omegaconf_available (line 291) | def is_omegaconf_available(): function is_tensorboard_available (line 295) | def is_tensorboard_available(): function requires_backends (line 389) | def requires_backends(obj, backends): class DummyObject (line 420) | class DummyObject(type): method __getattr__ (line 426) | def __getattr__(cls, key): function compare_versions (line 433) | def compare_versions(library_or_version: Union[str, Version], operation:... function is_torch_version (line 453) | def is_torch_version(operation: str, version: str): function is_transformers_version (line 465) | def is_transformers_version(operation: str, version: str): function is_accelerate_version (line 479) | def is_accelerate_version(operation: str, version: str): function is_k_diffusion_version (line 493) | def is_k_diffusion_version(operation: str, version: str): class OptionalDependencyNotAvailable (line 507) | class OptionalDependencyNotAvailable(BaseException): FILE: src/diffusers/utils/logging.py function _get_default_logging_level (line 52) | def _get_default_logging_level(): function _get_library_name (line 69) | def _get_library_name() -> str: function _get_library_root_logger (line 73) | def _get_library_root_logger() -> logging.Logger: function _configure_library_root_logger (line 77) | def _configure_library_root_logger() -> None: function _reset_library_root_logger (line 94) | def _reset_library_root_logger() -> None: function get_log_levels_dict (line 107) | def get_log_levels_dict(): function get_logger (line 111) | def get_logger(name: Optional[str] = None) -> logging.Logger: function get_verbosity (line 125) | def get_verbosity() -> int: function set_verbosity (line 148) | def set_verbosity(verbosity: int) -> None: function set_verbosity_info (line 167) | def set_verbosity_info(): function set_verbosity_warning (line 172) | def set_verbosity_warning(): function set_verbosity_debug (line 177) | def set_verbosity_debug(): function set_verbosity_error (line 182) | def set_verbosity_error(): function disable_default_handler (line 187) | def disable_default_handler() -> None: function enable_default_handler (line 196) | def enable_default_handler() -> None: function add_handler (line 205) | def add_handler(handler: logging.Handler) -> None: function remove_handler (line 214) | def remove_handler(handler: logging.Handler) -> None: function disable_propagation (line 223) | def disable_propagation() -> None: function enable_propagation (line 232) | def enable_propagation() -> None: function enable_explicit_format (line 242) | def enable_explicit_format() -> None: function reset_format (line 257) | def reset_format() -> None: function warning_advice (line 269) | def warning_advice(self, *args, **kwargs): class EmptyTqdm (line 283) | class EmptyTqdm: method __init__ (line 286) | def __init__(self, *args, **kwargs): # pylint: disable=unused-argument method __iter__ (line 289) | def __iter__(self): method __getattr__ (line 292) | def __getattr__(self, _): method __enter__ (line 300) | def __enter__(self): method __exit__ (line 303) | def __exit__(self, type_, value, traceback): class _tqdm_cls (line 307) | class _tqdm_cls: method __call__ (line 308) | def __call__(self, *args, **kwargs): method set_lock (line 314) | def set_lock(self, *args, **kwargs): method get_lock (line 319) | def get_lock(self): function is_progress_bar_enabled (line 327) | def is_progress_bar_enabled() -> bool: function enable_progress_bar (line 333) | def enable_progress_bar(): function disable_progress_bar (line 339) | def disable_progress_bar(): FILE: src/diffusers/utils/outputs.py function is_tensor (line 27) | def is_tensor(x): class BaseOutput (line 40) | class BaseOutput(OrderedDict): method __post_init__ (line 54) | def __post_init__(self): method __delitem__ (line 73) | def __delitem__(self, *args, **kwargs): method setdefault (line 76) | def setdefault(self, *args, **kwargs): method pop (line 79) | def pop(self, *args, **kwargs): method update (line 82) | def update(self, *args, **kwargs): method __getitem__ (line 85) | def __getitem__(self, k): method __setattr__ (line 92) | def __setattr__(self, name, value): method __setitem__ (line 98) | def __setitem__(self, key, value): method to_tuple (line 104) | def to_tuple(self) -> Tuple[Any]: FILE: src/diffusers/utils/testing_utils.py function torch_all_close (line 52) | def torch_all_close(a, b, *args, **kwargs): function print_tensor_test (line 60) | def print_tensor_test(tensor, filename="test_corrections.txt", expected_... function get_tests_dir (line 75) | def get_tests_dir(append_path=None): function parse_flag_from_env (line 96) | def parse_flag_from_env(key, default=False): function floats_tensor (line 116) | def floats_tensor(shape, scale=1.0, rng=None, name=None): function slow (line 132) | def slow(test_case): function nightly (line 142) | def nightly(test_case): function require_torch (line 152) | def require_torch(test_case): function require_torch_gpu (line 159) | def require_torch_gpu(test_case): function skip_mps (line 166) | def skip_mps(test_case): function require_flax (line 171) | def require_flax(test_case): function require_onnxruntime (line 178) | def require_onnxruntime(test_case): function load_numpy (line 185) | def load_numpy(arry: Union[str, np.ndarray], local_path: Optional[str] =... function load_pt (line 212) | def load_pt(url: str): function load_image (line 219) | def load_image(image: Union[str, PIL.Image.Image]) -> PIL.Image.Image: function load_hf_numpy (line 248) | def load_hf_numpy(path) -> np.ndarray: function pytest_addoption_shared (line 263) | def pytest_addoption_shared(parser): function pytest_terminal_summary_main (line 282) | def pytest_terminal_summary_main(tr, id): class CaptureLogger (line 411) | class CaptureLogger: method __init__ (line 432) | def __init__(self, logger): method __enter__ (line 438) | def __enter__(self): method __exit__ (line 442) | def __exit__(self, *exc): method __repr__ (line 446) | def __repr__(self): FILE: src/diffusers/utils/torch_utils.py function randn_tensor (line 29) | def randn_tensor( FILE: tests/conftest.py function pytest_addoption (line 33) | def pytest_addoption(parser): function pytest_terminal_summary (line 39) | def pytest_terminal_summary(terminalreporter): FILE: tests/fixtures/custom_pipeline/pipeline.py class CustomLocalPipeline (line 24) | class CustomLocalPipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet, scheduler): method __call__ (line 41) | def __call__( FILE: tests/fixtures/custom_pipeline/what_ever.py class CustomLocalPipeline (line 24) | class CustomLocalPipeline(DiffusionPipeline): method __init__ (line 36) | def __init__(self, unet, scheduler): method __call__ (line 41) | def __call__( FILE: tests/models/test_models_unet_1d.py class UNet1DModelTests (line 29) | class UNet1DModelTests(ModelTesterMixin, unittest.TestCase): method dummy_input (line 33) | def dummy_input(self): method input_shape (line 44) | def input_shape(self): method output_shape (line 48) | def output_shape(self): method test_ema_training (line 51) | def test_ema_training(self): method test_training (line 54) | def test_training(self): method test_determinism (line 58) | def test_determinism(self): method test_outputs_equivalence (line 62) | def test_outputs_equivalence(self): method test_from_save_pretrained (line 66) | def test_from_save_pretrained(self): method test_from_save_pretrained_variant (line 70) | def test_from_save_pretrained_variant(self): method test_model_from_pretrained (line 74) | def test_model_from_pretrained(self): method test_output (line 78) | def test_output(self): method prepare_init_args_and_inputs_for_common (line 81) | def prepare_init_args_and_inputs_for_common(self): method test_from_pretrained_hub (line 100) | def test_from_pretrained_hub(self): method test_output_pretrained (line 113) | def test_output_pretrained(self): method test_forward_with_norm_groups (line 135) | def test_forward_with_norm_groups(self): method test_unet_1d_maestro (line 140) | def test_unet_1d_maestro(self): class UNetRLModelTests (line 159) | class UNetRLModelTests(ModelTesterMixin, unittest.TestCase): method dummy_input (line 163) | def dummy_input(self): method input_shape (line 174) | def input_shape(self): method output_shape (line 178) | def output_shape(self): method test_determinism (line 182) | def test_determinism(self): method test_outputs_equivalence (line 186) | def test_outputs_equivalence(self): method test_from_save_pretrained (line 190) | def test_from_save_pretrained(self): method test_from_save_pretrained_variant (line 194) | def test_from_save_pretrained_variant(self): method test_model_from_pretrained (line 198) | def test_model_from_pretrained(self): method test_output (line 202) | def test_output(self): method test_ema_training (line 219) | def test_ema_training(self): method test_training (line 222) | def test_training(self): method prepare_init_args_and_inputs_for_common (line 225) | def prepare_init_args_and_inputs_for_common(self): method test_from_pretrained_hub (line 246) | def test_from_pretrained_hub(self): method test_output_pretrained (line 259) | def test_output_pretrained(self): method test_forward_with_norm_groups (line 282) | def test_forward_with_norm_groups(self): FILE: tests/models/test_models_unet_2d.py class Unet2DModelTests (line 33) | class Unet2DModelTests(ModelTesterMixin, unittest.TestCase): method dummy_input (line 37) | def dummy_input(self): method input_shape (line 48) | def input_shape(self): method output_shape (line 52) | def output_shape(self): method prepare_init_args_and_inputs_for_common (line 55) | def prepare_init_args_and_inputs_for_common(self): class UNetLDMModelTests (line 70) | class UNetLDMModelTests(ModelTesterMixin, unittest.TestCase): method dummy_input (line 74) | def dummy_input(self): method input_shape (line 85) | def input_shape(self): method output_shape (line 89) | def output_shape(self): method prepare_init_args_and_inputs_for_common (line 92) | def prepare_init_args_and_inputs_for_common(self): method test_from_pretrained_hub (line 106) | def test_from_pretrained_hub(self): method test_from_pretrained_accelerate (line 118) | def test_from_pretrained_accelerate(self): method test_from_pretrained_accelerate_wont_change_results (line 126) | def test_from_pretrained_accelerate_wont_change_results(self): method test_memory_footprint_gets_reduced (line 159) | def test_memory_footprint_gets_reduced(self): method test_output_pretrained (line 185) | def test_output_pretrained(self): class NCSNppModelTests (line 211) | class NCSNppModelTests(ModelTesterMixin, unittest.TestCase): method dummy_input (line 215) | def dummy_input(self, sizes=(32, 32)): method input_shape (line 225) | def input_shape(self): method output_shape (line 229) | def output_shape(self): method prepare_init_args_and_inputs_for_common (line 232) | def prepare_init_args_and_inputs_for_common(self): method test_from_pretrained_hub (line 259) | def test_from_pretrained_hub(self): method test_output_pretrained_ve_mid (line 273) | def test_output_pretrained_ve_mid(self): method test_output_pretrained_ve_large (line 298) | def test_output_pretrained_ve_large(self): method test_forward_with_norm_groups (line 323) | def test_forward_with_norm_groups(self): FILE: tests/models/test_models_unet_2d_condition.py function create_lora_layers (line 44) | def create_lora_layers(model): class UNet2DConditionModelTests (line 72) | class UNet2DConditionModelTests(ModelTesterMixin, unittest.TestCase): method dummy_input (line 76) | def dummy_input(self): method input_shape (line 88) | def input_shape(self): method output_shape (line 92) | def output_shape(self): method prepare_init_args_and_inputs_for_common (line 95) | def prepare_init_args_and_inputs_for_common(self): method test_xformers_enable_works (line 114) | def test_xformers_enable_works(self): method test_gradient_checkpointing (line 125) | def test_gradient_checkpointing(self): method test_model_with_attention_head_dim_tuple (line 165) | def test_model_with_attention_head_dim_tuple(self): method test_model_with_use_linear_projection (line 184) | def test_model_with_use_linear_projection(self): method test_model_attention_slicing (line 203) | def test_model_attention_slicing(self): method test_model_slicable_head_dim (line 227) | def test_model_slicable_head_dim(self): method test_special_attn_proc (line 245) | def test_special_attn_proc(self): method test_lora_processors (line 302) | def test_lora_processors(self): method test_lora_save_load (line 355) | def test_lora_save_load(self): method test_lora_save_load_safetensors (line 390) | def test_lora_save_load_safetensors(self): method test_lora_save_load_safetensors_load_torch (line 448) | def test_lora_save_load_safetensors_load_torch(self): method test_lora_on_off (line 485) | def test_lora_on_off(self): method test_lora_xformers_on_off (line 516) | def test_lora_xformers_on_off(self): class UNet2DConditionModelIntegrationTests (line 543) | class UNet2DConditionModelIntegrationTests(unittest.TestCase): method get_file_format (line 544) | def get_file_format(self, seed, shape): method tearDown (line 547) | def tearDown(self): method get_latents (line 553) | def get_latents(self, seed=0, shape=(4, 4, 64, 64), fp16=False): method get_unet_model (line 558) | def get_unet_model(self, fp16=False, model_id="CompVis/stable-diffusio... method test_set_attention_slice_auto (line 569) | def test_set_attention_slice_auto(self): method test_set_attention_slice_max (line 588) | def test_set_attention_slice_max(self): method test_set_attention_slice_int (line 607) | def test_set_attention_slice_int(self): method test_set_attention_slice_list (line 626) | def test_set_attention_slice_list(self): method get_encoder_hidden_states (line 647) | def get_encoder_hidden_states(self, seed=0, shape=(4, 77, 768), fp16=F... method test_compvis_sd_v1_4 (line 663) | def test_compvis_sd_v1_4(self, seed, timestep, expected_slice): method test_compvis_sd_v1_4_fp16 (line 691) | def test_compvis_sd_v1_4_fp16(self, seed, timestep, expected_slice): method test_compvis_sd_v1_5 (line 719) | def test_compvis_sd_v1_5(self, seed, timestep, expected_slice): method test_compvis_sd_v1_5_fp16 (line 747) | def test_compvis_sd_v1_5_fp16(self, seed, timestep, expected_slice): method test_compvis_sd_inpaint (line 775) | def test_compvis_sd_inpaint(self, seed, timestep, expected_slice): method test_compvis_sd_inpaint_fp16 (line 803) | def test_compvis_sd_inpaint_fp16(self, seed, timestep, expected_slice): method test_stabilityai_sd_v2_fp16 (line 831) | def test_stabilityai_sd_v2_fp16(self, seed, timestep, expected_slice): FILE: tests/models/test_models_unet_2d_flax.py class FlaxUNet2DConditionModelIntegrationTests (line 18) | class FlaxUNet2DConditionModelIntegrationTests(unittest.TestCase): method get_file_format (line 19) | def get_file_format(self, seed, shape): method tearDown (line 22) | def tearDown(self): method get_latents (line 27) | def get_latents(self, seed=0, shape=(4, 4, 64, 64), fp16=False): method get_unet_model (line 32) | def get_unet_model(self, fp16=False, model_id="CompVis/stable-diffusio... method get_encoder_hidden_states (line 41) | def get_encoder_hidden_states(self, seed=0, shape=(4, 77, 768), fp16=F... method test_compvis_sd_v1_4_flax_vs_torch_fp16 (line 56) | def test_compvis_sd_v1_4_flax_vs_torch_fp16(self, seed, timestep, expe... method test_stabilityai_sd_v2_flax_vs_torch_fp16 (line 86) | def test_stabilityai_sd_v2_flax_vs_torch_fp16(self, seed, timestep, ex... FILE: tests/models/test_models_vae.py class AutoencoderKLTests (line 32) | class AutoencoderKLTests(ModelTesterMixin, unittest.TestCase): method dummy_input (line 36) | def dummy_input(self): method input_shape (line 46) | def input_shape(self): method output_shape (line 50) | def output_shape(self): method prepare_init_args_and_inputs_for_common (line 53) | def prepare_init_args_and_inputs_for_common(self): method test_forward_signature (line 65) | def test_forward_signature(self): method test_training (line 68) | def test_training(self): method test_from_pretrained_hub (line 71) | def test_from_pretrained_hub(self): method test_output_pretrained (line 81) | def test_output_pretrained(self): class AutoencoderKLIntegrationTests (line 138) | class AutoencoderKLIntegrationTests(unittest.TestCase): method get_file_format (line 139) | def get_file_format(self, seed, shape): method tearDown (line 142) | def tearDown(self): method get_sd_image (line 148) | def get_sd_image(self, seed=0, shape=(4, 3, 512, 512), fp16=False): method get_sd_vae_model (line 153) | def get_sd_vae_model(self, model_id="CompVis/stable-diffusion-v1-4", f... method get_generator (line 167) | def get_generator(self, seed=0): method test_stable_diffusion (line 180) | def test_stable_diffusion(self, seed, expected_slice, expected_slice_m... method test_stable_diffusion_fp16 (line 204) | def test_stable_diffusion_fp16(self, seed, expected_slice): method test_stable_diffusion_mode (line 227) | def test_stable_diffusion_mode(self, seed, expected_slice, expected_sl... method test_stable_diffusion_decode (line 250) | def test_stable_diffusion_decode(self, seed, expected_slice): method test_stable_diffusion_decode_fp16 (line 273) | def test_stable_diffusion_decode_fp16(self, seed, expected_slice): method test_stable_diffusion_encode_sample (line 295) | def test_stable_diffusion_encode_sample(self, seed, expected_slice): FILE: tests/models/test_models_vae_flax.py class FlaxAutoencoderKLTests (line 15) | class FlaxAutoencoderKLTests(FlaxModelTesterMixin, unittest.TestCase): method dummy_input (line 19) | def dummy_input(self): method prepare_init_args_and_inputs_for_common (line 29) | def prepare_init_args_and_inputs_for_common(self): FILE: tests/models/test_models_vq.py class VQModelTests (line 29) | class VQModelTests(ModelTesterMixin, unittest.TestCase): method dummy_input (line 33) | def dummy_input(self, sizes=(32, 32)): method input_shape (line 42) | def input_shape(self): method output_shape (line 46) | def output_shape(self): method prepare_init_args_and_inputs_for_common (line 49) | def prepare_init_args_and_inputs_for_common(self): method test_forward_signature (line 61) | def test_forward_signature(self): method test_training (line 64) | def test_training(self): method test_from_pretrained_hub (line 67) | def test_from_pretrained_hub(self): method test_output_pretrained (line 77) | def test_output_pretrained(self): FILE: tests/pipelines/altdiffusion/test_alt_diffusion.py class AltDiffusionPipelineFastTests (line 38) | class AltDiffusionPipelineFastTests(PipelineTesterMixin, unittest.TestCa... method get_dummy_components (line 43) | def get_dummy_components(self): method get_dummy_inputs (line 114) | def get_dummy_inputs(self, device, seed=0): method test_alt_diffusion_ddim (line 128) | def test_alt_diffusion_ddim(self): method test_alt_diffusion_pndm (line 163) | def test_alt_diffusion_pndm(self): class AltDiffusionPipelineIntegrationTests (line 200) | class AltDiffusionPipelineIntegrationTests(unittest.TestCase): method tearDown (line 201) | def tearDown(self): method test_alt_diffusion (line 207) | def test_alt_diffusion(self): method test_alt_diffusion_fast_ddim (line 226) | def test_alt_diffusion_fast_ddim(self): FILE: tests/pipelines/altdiffusion/test_alt_diffusion_img2img.py class AltDiffusionImg2ImgPipelineFastTests (line 36) | class AltDiffusionImg2ImgPipelineFastTests(unittest.TestCase): method tearDown (line 37) | def tearDown(self): method dummy_image (line 44) | def dummy_image(self): method dummy_cond_unet (line 53) | def dummy_cond_unet(self): method dummy_vae (line 68) | def dummy_vae(self): method dummy_text_encoder (line 81) | def dummy_text_encoder(self): method dummy_extractor (line 96) | def dummy_extractor(self): method test_stable_diffusion_img2img_default_case (line 110) | def test_stable_diffusion_img2img_default_case(self): method test_stable_diffusion_img2img_fp16 (line 168) | def test_stable_diffusion_img2img_fp16(self): method test_stable_diffusion_img2img_pipeline_multiple_of_8 (line 210) | def test_stable_diffusion_img2img_pipeline_multiple_of_8(self): class AltDiffusionImg2ImgPipelineIntegrationTests (line 250) | class AltDiffusionImg2ImgPipelineIntegrationTests(unittest.TestCase): method tearDown (line 251) | def tearDown(self): method test_stable_diffusion_img2img_pipeline_default (line 257) | def test_stable_diffusion_img2img_pipeline_default(self): FILE: tests/pipelines/audio_diffusion/test_audio_diffusion.py class PipelineFastTests (line 39) | class PipelineFastTests(unittest.TestCase): method tearDown (line 40) | def tearDown(self): method dummy_unet (line 47) | def dummy_unet(self): method dummy_unet_condition (line 61) | def dummy_unet_condition(self): method dummy_vqvae_and_unet (line 76) | def dummy_vqvae_and_unet(self): method test_audio_diffusion (line 100) | def test_audio_diffusion(self): class PipelineIntegrationTests (line 167) | class PipelineIntegrationTests(unittest.TestCase): method tearDown (line 168) | def tearDown(self): method test_audio_diffusion (line 174) | def test_audio_diffusion(self): FILE: tests/pipelines/dance_diffusion/test_dance_diffusion.py class DanceDiffusionPipelineFastTests (line 33) | class DanceDiffusionPipelineFastTests(PipelineTesterMixin, unittest.Test... method get_dummy_components (line 47) | def get_dummy_components(self): method get_dummy_inputs (line 71) | def get_dummy_inputs(self, device, seed=0): method test_dance_diffusion (line 83) | def test_dance_diffusion(self): method test_save_load_local (line 101) | def test_save_load_local(self): method test_dict_tuple_outputs_equivalent (line 105) | def test_dict_tuple_outputs_equivalent(self): method test_save_load_optional_components (line 109) | def test_save_load_optional_components(self): method test_attention_slicing_forward_pass (line 113) | def test_attention_slicing_forward_pass(self): class PipelineIntegrationTests (line 119) | class PipelineIntegrationTests(unittest.TestCase): method tearDown (line 120) | def tearDown(self): method test_dance_diffusion (line 126) | def test_dance_diffusion(self): method test_dance_diffusion_fp16 (line 144) | def test_dance_diffusion_fp16(self): FILE: tests/pipelines/ddim/test_ddim.py class DDIMPipelineFastTests (line 31) | class DDIMPipelineFastTests(PipelineTesterMixin, unittest.TestCase): method get_dummy_components (line 43) | def get_dummy_components(self): method get_dummy_inputs (line 58) | def get_dummy_inputs(self, device, seed=0): method test_inference (line 71) | def test_inference(self): class DDIMPipelineIntegrationTests (line 93) | class DDIMPipelineIntegrationTests(unittest.TestCase): method test_inference_cifar10 (line 94) | def test_inference_cifar10(self): method test_inference_ema_bedroom (line 114) | def test_inference_ema_bedroom(self): FILE: tests/pipelines/ddpm/test_ddpm.py class DDPMPipelineFastTests (line 28) | class DDPMPipelineFastTests(unittest.TestCase): method dummy_uncond_unet (line 30) | def dummy_uncond_unet(self): method test_fast_inference (line 43) | def test_fast_inference(self): method test_inference_predict_sample (line 69) | def test_inference_predict_sample(self): class DDPMPipelineIntegrationTests (line 97) | class DDPMPipelineIntegrationTests(unittest.TestCase): method test_inference_cifar10 (line 98) | def test_inference_cifar10(self): FILE: tests/pipelines/dit/test_dit.py class DiTPipelineFastTests (line 36) | class DiTPipelineFastTests(PipelineTesterMixin, unittest.TestCase): method get_dummy_components (line 48) | def get_dummy_components(self): method get_dummy_inputs (line 69) | def get_dummy_inputs(self, device, seed=0): method test_inference (line 82) | def test_inference(self): method test_inference_batch_single_identical (line 99) | def test_inference_batch_single_identical(self): class DiTPipelineIntegrationTests (line 105) | class DiTPipelineIntegrationTests(unittest.TestCase): method tearDown (line 106) | def tearDown(self): method test_dit_256 (line 111) | def test_dit_256(self): method test_dit_512_fp16 (line 128) | def test_dit_512_fp16(self): FILE: tests/pipelines/karras_ve/test_karras_ve.py class KarrasVePipelineFastTests (line 28) | class KarrasVePipelineFastTests(unittest.TestCase): method dummy_uncond_unet (line 30) | def dummy_uncond_unet(self): method test_inference (line 43) | def test_inference(self): class KarrasVePipelineIntegrationTests (line 69) | class KarrasVePipelineIntegrationTests(unittest.TestCase): method test_inference (line 70) | def test_inference(self): FILE: tests/pipelines/latent_diffusion/test_latent_diffusion.py class LDMTextToImagePipelineFastTests (line 33) | class LDMTextToImagePipelineFastTests(PipelineTesterMixin, unittest.Test... method get_dummy_components (line 49) | def get_dummy_components(self): method get_dummy_inputs (line 101) | def get_dummy_inputs(self, device, seed=0): method test_inference_text2img (line 115) | def test_inference_text2img(self): class LDMTextToImagePipelineSlowTests (line 135) | class LDMTextToImagePipelineSlowTests(unittest.TestCase): method tearDown (line 136) | def tearDown(self): method get_inputs (line 141) | def get_inputs(self, device, dtype=torch.float32, seed=0): method test_ldm_default_ddim (line 155) | def test_ldm_default_ddim(self): class LDMTextToImagePipelineNightlyTests (line 171) | class LDMTextToImagePipelineNightlyTests(unittest.TestCase): method tearDown (line 172) | def tearDown(self): method get_inputs (line 177) | def get_inputs(self, device, dtype=torch.float32, seed=0): method test_ldm_default_ddim (line 191) | def test_ldm_default_ddim(self): FILE: tests/pipelines/latent_diffusion/test_latent_diffusion_superresolution.py class LDMSuperResolutionPipelineFastTests (line 30) | class LDMSuperResolutionPipelineFastTests(unittest.TestCase): method dummy_image (line 32) | def dummy_image(self): method dummy_uncond_unet (line 41) | def dummy_uncond_unet(self): method dummy_vq_model (line 55) | def dummy_vq_model(self): method test_inference_superresolution (line 67) | def test_inference_superresolution(self): method test_inference_superresolution_fp16 (line 90) | def test_inference_superresolution_fp16(self): class LDMSuperResolutionPipelineIntegrationTests (line 112) | class LDMSuperResolutionPipelineIntegrationTests(unittest.TestCase): method test_inference_superresolution (line 113) | def test_inference_superresolution(self): FILE: tests/pipelines/latent_diffusion/test_latent_diffusion_uncond.py class LDMPipelineFastTests (line 29) | class LDMPipelineFastTests(unittest.TestCase): method dummy_uncond_unet (line 31) | def dummy_uncond_unet(self): method dummy_vq_model (line 45) | def dummy_vq_model(self): method dummy_text_encoder (line 58) | def dummy_text_encoder(self): method test_inference_uncond (line 73) | def test_inference_uncond(self): class LDMPipelineIntegrationTests (line 106) | class LDMPipelineIntegrationTests(unittest.TestCase): method test_inference_uncond (line 107) | def test_inference_uncond(self): FILE: tests/pipelines/paint_by_example/test_paint_by_example.py class PaintByExamplePipelineFastTests (line 37) | class PaintByExamplePipelineFastTests(PipelineTesterMixin, unittest.Test... method get_dummy_components (line 42) | def get_dummy_components(self): method convert_to_pt (line 88) | def convert_to_pt(self, image): method get_dummy_inputs (line 94) | def get_dummy_inputs(self, device="cpu", seed=0): method test_paint_by_example_inpaint (line 117) | def test_paint_by_example_inpaint(self): method test_paint_by_example_image_tensor (line 136) | def test_paint_by_example_image_tensor(self): class PaintByExamplePipelineIntegrationTests (line 166) | class PaintByExamplePipelineIntegrationTests(unittest.TestCase): method tearDown (line 167) | def tearDown(self): method test_paint_by_example (line 173) | def test_paint_by_example(self): FILE: tests/pipelines/pndm/test_pndm.py class PNDMPipelineFastTests (line 28) | class PNDMPipelineFastTests(unittest.TestCase): method dummy_uncond_unet (line 30) | def dummy_uncond_unet(self): method test_inference (line 43) | def test_inference(self): class PNDMPipelineIntegrationTests (line 69) | class PNDMPipelineIntegrationTests(unittest.TestCase): method test_inference_cifar10 (line 70) | def test_inference_cifar10(self): FILE: tests/pipelines/repaint/test_repaint.py class RepaintPipelineFastTests (line 32) | class RepaintPipelineFastTests(PipelineTesterMixin, unittest.TestCase): method get_dummy_components (line 44) | def get_dummy_components(self): method get_dummy_inputs (line 60) | def get_dummy_inputs(self, device, seed=0): method test_repaint (line 80) | def test_repaint(self): method test_save_load_local (line 97) | def test_save_load_local(self): method test_inference_batch_single_identical (line 103) | def test_inference_batch_single_identical(self): method test_dict_tuple_outputs_equivalent (line 107) | def test_dict_tuple_outputs_equivalent(self): method test_save_load_optional_components (line 111) | def test_save_load_optional_components(self): method test_attention_slicing_forward_pass (line 115) | def test_attention_slicing_forward_pass(self): class RepaintPipelineNightlyTests (line 121) | class RepaintPipelineNightlyTests(unittest.TestCase): method tearDown (line 122) | def tearDown(self): method test_celebahq (line 127) | def test_celebahq(self): FILE: tests/pipelines/score_sde_ve/test_score_sde_ve.py class ScoreSdeVeipelineFastTests (line 28) | class ScoreSdeVeipelineFastTests(unittest.TestCase): method dummy_uncond_unet (line 30) | def dummy_uncond_unet(self): method test_inference (line 43) | def test_inference(self): class ScoreSdeVePipelineIntegrationTests (line 71) | class ScoreSdeVePipelineIntegrationTests(unittest.TestCase): method test_inference (line 72) | def test_inference(self): FILE: tests/pipelines/semantic_stable_diffusion/test_semantic_diffusion.py class SafeDiffusionPipelineFastTests (line 34) | class SafeDiffusionPipelineFastTests(unittest.TestCase): method tearDown (line 35) | def tearDown(self): method dummy_image (line 42) | def dummy_image(self): method dummy_cond_unet (line 51) | def dummy_cond_unet(self): method dummy_vae (line 66) | def dummy_vae(self): method dummy_text_encoder (line 79) | def dummy_text_encoder(self): method dummy_extractor (line 95) | def dummy_extractor(self): method test_semantic_diffusion_ddim (line 109) | def test_semantic_diffusion_ddim(self): method test_semantic_diffusion_pndm (line 162) | def test_semantic_diffusion_pndm(self): method test_semantic_diffusion_no_safety_checker (line 208) | def test_semantic_diffusion_no_safety_checker(self): method test_semantic_diffusion_fp16 (line 230) | def test_semantic_diffusion_fp16(self): class SemanticDiffusionPipelineIntegrationTests (line 264) | class SemanticDiffusionPipelineIntegrationTests(unittest.TestCase): method tearDown (line 265) | def tearDown(self): method test_positive_guidance (line 271) | def test_positive_guidance(self): method test_negative_guidance (line 354) | def test_negative_guidance(self): method test_multi_cond_guidance (line 437) | def test_multi_cond_guidance(self): method test_guidance_fp16 (line 520) | def test_guidance_fp16(self): FILE: tests/pipelines/stable_diffusion/test_cycle_diffusion.py class CycleDiffusionPipelineFastTests (line 35) | class CycleDiffusionPipelineFastTests(PipelineTesterMixin, unittest.Test... method get_dummy_components (line 46) | def get_dummy_components(self): method get_dummy_inputs (line 101) | def get_dummy_inputs(self, device, seed=0): method test_stable_diffusion_cycle (line 121) | def test_stable_diffusion_cycle(self): method test_stable_diffusion_cycle_fp16 (line 141) | def test_stable_diffusion_cycle_fp16(self): method test_save_load_local (line 162) | def test_save_load_local(self): method test_inference_batch_single_identical (line 166) | def test_inference_batch_single_identical(self): method test_dict_tuple_outputs_equivalent (line 170) | def test_dict_tuple_outputs_equivalent(self): method test_save_load_optional_components (line 174) | def test_save_load_optional_components(self): method test_attention_slicing_forward_pass (line 178) | def test_attention_slicing_forward_pass(self): class CycleDiffusionPipelineIntegrationTests (line 184) | class CycleDiffusionPipelineIntegrationTests(unittest.TestCase): method tearDown (line 185) | def tearDown(self): method test_cycle_diffusion_pipeline_fp16 (line 191) | def test_cycle_diffusion_pipeline_fp16(self): method test_cycle_diffusion_pipeline (line 232) | def test_cycle_diffusion_pipeline(self): FILE: tests/pipelines/stable_diffusion/test_onnx_stable_diffusion.py class OnnxStableDiffusionPipelineFastTests (line 39) | class OnnxStableDiffusionPipelineFastTests(OnnxPipelineTesterMixin, unit... method get_dummy_inputs (line 42) | def get_dummy_inputs(self, seed=0): method test_pipeline_default_ddim (line 53) | def test_pipeline_default_ddim(self): method test_pipeline_pndm (line 66) | def test_pipeline_pndm(self): method test_pipeline_lms (line 80) | def test_pipeline_lms(self): method test_pipeline_euler (line 94) | def test_pipeline_euler(self): method test_pipeline_euler_ancestral (line 108) | def test_pipeline_euler_ancestral(self): method test_pipeline_dpm_multistep (line 122) | def test_pipeline_dpm_multistep(self): class OnnxStableDiffusionPipelineIntegrationTests (line 140) | class OnnxStableDiffusionPipelineIntegrationTests(unittest.TestCase): method gpu_provider (line 142) | def gpu_provider(self): method gpu_options (line 152) | def gpu_options(self): method test_inference_default_pndm (line 157) | def test_inference_default_pndm(self): method test_inference_ddim (line 181) | def test_inference_ddim(self): method test_inference_k_lms (line 207) | def test_inference_k_lms(self): method test_intermediate_state (line 233) | def test_intermediate_state(self): method test_stable_diffusion_no_safety_checker (line 283) | def test_stable_diffusion_no_safety_checker(self): FILE: tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_img2img.py class OnnxStableDiffusionImg2ImgPipelineFastTests (line 45) | class OnnxStableDiffusionImg2ImgPipelineFastTests(OnnxPipelineTesterMixi... method get_dummy_inputs (line 48) | def get_dummy_inputs(self, seed=0): method test_pipeline_default_ddim (line 62) | def test_pipeline_default_ddim(self): method test_pipeline_pndm (line 74) | def test_pipeline_pndm(self): method test_pipeline_lms (line 88) | def test_pipeline_lms(self): method test_pipeline_euler (line 105) | def test_pipeline_euler(self): method test_pipeline_euler_ancestral (line 119) | def test_pipeline_euler_ancestral(self): method test_pipeline_dpm_multistep (line 133) | def test_pipeline_dpm_multistep(self): class OnnxStableDiffusionImg2ImgPipelineIntegrationTests (line 151) | class OnnxStableDiffusionImg2ImgPipelineIntegrationTests(unittest.TestCa... method gpu_provider (line 153) | def gpu_provider(self): method gpu_options (line 163) | def gpu_options(self): method test_inference_default_pndm (line 168) | def test_inference_default_pndm(self): method test_inference_k_lms (line 206) | def test_inference_k_lms(self): FILE: tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_inpaint.py class OnnxStableDiffusionPipelineFastTests (line 36) | class OnnxStableDiffusionPipelineFastTests(OnnxPipelineTesterMixin, unit... class OnnxStableDiffusionInpaintPipelineIntegrationTests (line 44) | class OnnxStableDiffusionInpaintPipelineIntegrationTests(unittest.TestCa... method gpu_provider (line 46) | def gpu_provider(self): method gpu_options (line 56) | def gpu_options(self): method test_inference_default_pndm (line 61) | def test_inference_default_pndm(self): method test_inference_k_lms (line 100) | def test_inference_k_lms(self): FILE: tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_inpaint_legacy.py class StableDiffusionOnnxInpaintLegacyPipelineIntegrationTests (line 38) | class StableDiffusionOnnxInpaintLegacyPipelineIntegrationTests(unittest.... method gpu_provider (line 40) | def gpu_provider(self): method gpu_options (line 50) | def gpu_options(self): method test_inference (line 55) | def test_inference(self): FILE: tests/pipelines/stable_diffusion/test_stable_diffusion.py class StableDiffusionPipelineFastTests (line 49) | class StableDiffusionPipelineFastTests(PipelineTesterMixin, unittest.Tes... method get_dummy_components (line 54) | def get_dummy_components(self): method get_dummy_inputs (line 108) | def get_dummy_inputs(self, device, seed=0): method test_stable_diffusion_ddim (line 122) | def test_stable_diffusion_ddim(self): method test_stable_diffusion_lora (line 141) | def test_stable_diffusion_lora(self): method test_stable_diffusion_prompt_embeds (line 175) | def test_stable_diffusion_prompt_embeds(self): method test_stable_diffusion_negative_prompt_embeds (line 211) | def test_stable_diffusion_negative_prompt_embeds(self): method test_stable_diffusion_ddim_factor_8 (line 251) | def test_stable_diffusion_ddim_factor_8(self): method test_stable_diffusion_pndm (line 270) | def test_stable_diffusion_pndm(self): method test_stable_diffusion_no_safety_checker (line 288) | def test_stable_diffusion_no_safety_checker(self): method test_stable_diffusion_k_lms (line 309) | def test_stable_diffusion_k_lms(self): method test_stable_diffusion_k_euler_ancestral (line 340) | def test_stable_diffusion_k_euler_ancestral(self): method test_stable_diffusion_k_euler (line 371) | def test_stable_diffusion_k_euler(self): method test_stable_diffusion_vae_slicing (line 402) | def test_stable_diffusion_vae_slicing(self): method test_stable_diffusion_vae_tiling (line 425) | def test_stable_diffusion_vae_tiling(self): method test_stable_diffusion_negative_prompt (line 448) | def test_stable_diffusion_negative_prompt(self): method test_stable_diffusion_long_prompt (line 480) | def test_stable_diffusion_long_prompt(self): method test_stable_diffusion_height_width_opt (line 518) | def test_stable_diffusion_height_width_opt(self): class StableDiffusionPipelineSlowTests (line 545) | class StableDiffusionPipelineSlowTests(unittest.TestCase): method tearDown (line 546) | def tearDown(self): method get_inputs (line 551) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_stable_diffusion_1_1_pndm (line 565) | def test_stable_diffusion_1_1_pndm(self): method test_stable_diffusion_1_4_pndm (line 578) | def test_stable_diffusion_1_4_pndm(self): method test_stable_diffusion_ddim (line 591) | def test_stable_diffusion_ddim(self): method test_stable_diffusion_lms (line 605) | def test_stable_diffusion_lms(self): method test_stable_diffusion_dpm (line 619) | def test_stable_diffusion_dpm(self): method test_stable_diffusion_attention_slicing (line 633) | def test_stable_diffusion_attention_slicing(self): method test_stable_diffusion_vae_slicing (line 659) | def test_stable_diffusion_vae_slicing(self): method test_stable_diffusion_vae_tiling (line 691) | def test_stable_diffusion_vae_tiling(self): method test_stable_diffusion_fp16_vs_autocast (line 743) | def test_stable_diffusion_fp16_vs_autocast(self): method test_stable_diffusion_intermediate_state (line 763) | def test_stable_diffusion_intermediate_state(self): method test_stable_diffusion_low_cpu_mem_usage (line 801) | def test_stable_diffusion_low_cpu_mem_usage(self): method test_stable_diffusion_pipeline_with_sequential_cpu_offloading (line 815) | def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): method test_stable_diffusion_pipeline_with_model_offloading (line 833) | def test_stable_diffusion_pipeline_with_model_offloading(self): class StableDiffusionPipelineNightlyTests (line 889) | class StableDiffusionPipelineNightlyTests(unittest.TestCase): method tearDown (line 890) | def tearDown(self): method get_inputs (line 895) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_stable_diffusion_1_4_pndm (line 909) | def test_stable_diffusion_1_4_pndm(self): method test_stable_diffusion_1_5_pndm (line 923) | def test_stable_diffusion_1_5_pndm(self): method test_stable_diffusion_ddim (line 937) | def test_stable_diffusion_ddim(self): method test_stable_diffusion_lms (line 952) | def test_stable_diffusion_lms(self): method test_stable_diffusion_euler (line 967) | def test_stable_diffusion_euler(self): method test_stable_diffusion_dpm (line 982) | def test_stable_diffusion_dpm(self): FILE: tests/pipelines/stable_diffusion/test_stable_diffusion_controlnet.py class StableDiffusionControlNetPipelineFastTests (line 38) | class StableDiffusionControlNetPipelineFastTests(PipelineTesterMixin, un... method get_dummy_components (line 43) | def get_dummy_components(self): method get_dummy_inputs (line 108) | def get_dummy_inputs(self, device, seed=0): method test_attention_slicing_forward_pass (line 132) | def test_attention_slicing_forward_pass(self): method test_xformers_attention_forwardGenerator_pass (line 139) | def test_xformers_attention_forwardGenerator_pass(self): method test_inference_batch_single_identical (line 142) | def test_inference_batch_single_identical(self): class StableDiffusionControlNetPipelineSlowTests (line 148) | class StableDiffusionControlNetPipelineSlowTests(unittest.TestCase): method tearDown (line 149) | def tearDown(self): method test_canny (line 154) | def test_canny(self): method test_depth (line 181) | def test_depth(self): method test_hed (line 208) | def test_hed(self): method test_mlsd (line 235) | def test_mlsd(self): method test_normal (line 262) | def test_normal(self): method test_openpose (line 289) | def test_openpose(self): method test_scribble (line 316) | def test_scribble(self): method test_seg (line 343) | def test_seg(self): method test_sequential_cpu_offloading (line 370) | def test_sequential_cpu_offloading(self): FILE: tests/pipelines/stable_diffusion/test_stable_diffusion_image_variation.py class StableDiffusionImageVariationPipelineFastTests (line 42) | class StableDiffusionImageVariationPipelineFastTests(PipelineTesterMixin... method get_dummy_components (line 47) | def get_dummy_components(self): method get_dummy_inputs (line 93) | def get_dummy_inputs(self, device, seed=0): method test_stable_diffusion_img_variation_default_case (line 110) | def test_stable_diffusion_img_variation_default_case(self): method test_stable_diffusion_img_variation_multiple_images (line 126) | def test_stable_diffusion_img_variation_multiple_images(self): class StableDiffusionImageVariationPipelineSlowTests (line 149) | class StableDiffusionImageVariationPipelineSlowTests(unittest.TestCase): method tearDown (line 150) | def tearDown(self): method get_inputs (line 155) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_stable_diffusion_img_variation_pipeline_default (line 173) | def test_stable_diffusion_img_variation_pipeline_default(self): method test_stable_diffusion_img_variation_intermediate_state (line 188) | def test_stable_diffusion_img_variation_intermediate_state(self): method test_stable_diffusion_pipeline_with_sequential_cpu_offloading (line 228) | def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): class StableDiffusionImageVariationPipelineNightlyTests (line 252) | class StableDiffusionImageVariationPipelineNightlyTests(unittest.TestCase): method tearDown (line 253) | def tearDown(self): method get_inputs (line 258) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_img_variation_pndm (line 276) | def test_img_variation_pndm(self): method test_img_variation_dpm (line 291) | def test_img_variation_dpm(self): FILE: tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py class StableDiffusionImg2ImgPipelineFastTests (line 43) | class StableDiffusionImg2ImgPipelineFastTests(PipelineTesterMixin, unitt... method get_dummy_components (line 49) | def get_dummy_components(self): method get_dummy_inputs (line 97) | def get_dummy_inputs(self, device, seed=0): method test_stable_diffusion_img2img_default_case (line 113) | def test_stable_diffusion_img2img_default_case(self): method test_stable_diffusion_img2img_negative_prompt (line 129) | def test_stable_diffusion_img2img_negative_prompt(self): method test_stable_diffusion_img2img_multiple_init_images (line 147) | def test_stable_diffusion_img2img_multiple_init_images(self): method test_stable_diffusion_img2img_k_lms (line 165) | def test_stable_diffusion_img2img_k_lms(self): method test_save_load_local (line 185) | def test_save_load_local(self): method test_dict_tuple_outputs_equivalent (line 189) | def test_dict_tuple_outputs_equivalent(self): method test_save_load_optional_components (line 193) | def test_save_load_optional_components(self): method test_attention_slicing_forward_pass (line 197) | def test_attention_slicing_forward_pass(self): class StableDiffusionImg2ImgPipelineSlowTests (line 203) | class StableDiffusionImg2ImgPipelineSlowTests(unittest.TestCase): method tearDown (line 204) | def tearDown(self): method get_inputs (line 209) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_stable_diffusion_img2img_default (line 226) | def test_stable_diffusion_img2img_default(self): method test_stable_diffusion_img2img_k_lms (line 241) | def test_stable_diffusion_img2img_k_lms(self): method test_stable_diffusion_img2img_ddim (line 257) | def test_stable_diffusion_img2img_ddim(self): method test_stable_diffusion_img2img_intermediate_state (line 273) | def test_stable_diffusion_img2img_intermediate_state(self): method test_stable_diffusion_pipeline_with_sequential_cpu_offloading (line 309) | def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): method test_stable_diffusion_pipeline_with_model_offloading (line 329) | def test_stable_diffusion_pipeline_with_model_offloading(self): method test_stable_diffusion_img2img_pipeline_multiple_of_8 (line 370) | def test_stable_diffusion_img2img_pipeline_multiple_of_8(self): class StableDiffusionImg2ImgPipelineNightlyTests (line 410) | class StableDiffusionImg2ImgPipelineNightlyTests(unittest.TestCase): method tearDown (line 411) | def tearDown(self): method get_inputs (line 416) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_img2img_pndm (line 433) | def test_img2img_pndm(self): method test_img2img_ddim (line 448) | def test_img2img_ddim(self): method test_img2img_lms (line 464) | def test_img2img_lms(self): method test_img2img_dpm (line 480) | def test_img2img_dpm(self): FILE: tests/pipelines/stable_diffusion/test_stable_diffusion_inpaint.py class StableDiffusionInpaintPipelineFastTests (line 44) | class StableDiffusionInpaintPipelineFastTests(PipelineTesterMixin, unitt... method get_dummy_components (line 49) | def get_dummy_components(self): method get_dummy_inputs (line 97) | def get_dummy_inputs(self, device, seed=0): method test_stable_diffusion_inpaint (line 118) | def test_stable_diffusion_inpaint(self): method test_stable_diffusion_inpaint_image_tensor (line 134) | def test_stable_diffusion_inpaint_image_tensor(self): class StableDiffusionInpaintPipelineSlowTests (line 157) | class StableDiffusionInpaintPipelineSlowTests(unittest.TestCase): method setUp (line 158) | def setUp(self): method tearDown (line 161) | def tearDown(self): method get_inputs (line 166) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_stable_diffusion_inpaint_ddim (line 187) | def test_stable_diffusion_inpaint_ddim(self): method test_stable_diffusion_inpaint_fp16 (line 204) | def test_stable_diffusion_inpaint_fp16(self): method test_stable_diffusion_inpaint_pndm (line 221) | def test_stable_diffusion_inpaint_pndm(self): method test_stable_diffusion_inpaint_k_lms (line 239) | def test_stable_diffusion_inpaint_k_lms(self): method test_stable_diffusion_inpaint_with_sequential_cpu_offloading (line 257) | def test_stable_diffusion_inpaint_with_sequential_cpu_offloading(self): class StableDiffusionInpaintPipelineNightlyTests (line 280) | class StableDiffusionInpaintPipelineNightlyTests(unittest.TestCase): method tearDown (line 281) | def tearDown(self): method get_inputs (line 286) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_inpaint_ddim (line 307) | def test_inpaint_ddim(self): method test_inpaint_pndm (line 322) | def test_inpaint_pndm(self): method test_inpaint_lms (line 338) | def test_inpaint_lms(self): method test_inpaint_dpm (line 354) | def test_inpaint_dpm(self): class StableDiffusionInpaintingPrepareMaskAndMaskedImageTests (line 372) | class StableDiffusionInpaintingPrepareMaskAndMaskedImageTests(unittest.T... method test_pil_inputs (line 373) | def test_pil_inputs(self): method test_np_inputs (line 400) | def test_np_inputs(self): method test_torch_3D_2D_inputs (line 412) | def test_torch_3D_2D_inputs(self): method test_torch_3D_3D_inputs (line 424) | def test_torch_3D_3D_inputs(self): method test_torch_4D_2D_inputs (line 436) | def test_torch_4D_2D_inputs(self): method test_torch_4D_3D_inputs (line 448) | def test_torch_4D_3D_inputs(self): method test_torch_4D_4D_inputs (line 460) | def test_torch_4D_4D_inputs(self): method test_torch_batch_4D_3D (line 472) | def test_torch_batch_4D_3D(self): method test_torch_batch_4D_4D (line 487) | def test_torch_batch_4D_4D(self): method test_shape_mismatch (line 502) | def test_shape_mismatch(self): method test_type_mismatch (line 513) | def test_type_mismatch(self): method test_channels_first (line 521) | def test_channels_first(self): method test_tensor_range (line 526) | def test_tensor_range(self): FILE: tests/pipelines/stable_diffusion/test_stable_diffusion_inpaint_legacy.py class StableDiffusionInpaintLegacyPipelineFastTests (line 43) | class StableDiffusionInpaintLegacyPipelineFastTests(unittest.TestCase): method tearDown (line 44) | def tearDown(self): method dummy_image (line 51) | def dummy_image(self): method dummy_uncond_unet (line 60) | def dummy_uncond_unet(self): method dummy_cond_unet (line 74) | def dummy_cond_unet(self): method dummy_cond_unet_inpaint (line 89) | def dummy_cond_unet_inpaint(self): method dummy_vq_model (line 104) | def dummy_vq_model(self): method dummy_vae (line 117) | def dummy_vae(self): method dummy_text_encoder (line 130) | def dummy_text_encoder(self): method dummy_extractor (line 146) | def dummy_extractor(self): method test_stable_diffusion_inpaint_legacy (line 160) | def test_stable_diffusion_inpaint_legacy(self): method test_stable_diffusion_inpaint_legacy_negative_prompt (line 220) | def test_stable_diffusion_inpaint_legacy_negative_prompt(self): method test_stable_diffusion_inpaint_legacy_num_images_per_prompt (line 267) | def test_stable_diffusion_inpaint_legacy_num_images_per_prompt(self): class StableDiffusionInpaintLegacyPipelineSlowTests (line 346) | class StableDiffusionInpaintLegacyPipelineSlowTests(unittest.TestCase): method tearDown (line 347) | def tearDown(self): method get_inputs (line 352) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_stable_diffusion_inpaint_legacy_pndm (line 374) | def test_stable_diffusion_inpaint_legacy_pndm(self): method test_stable_diffusion_inpaint_legacy_k_lms (line 391) | def test_stable_diffusion_inpaint_legacy_k_lms(self): method test_stable_diffusion_inpaint_legacy_intermediate_state (line 409) | def test_stable_diffusion_inpaint_legacy_intermediate_state(self): class StableDiffusionInpaintLegacyPipelineNightlyTests (line 448) | class StableDiffusionInpaintLegacyPipelineNightlyTests(unittest.TestCase): method tearDown (line 449) | def tearDown(self): method get_inputs (line 454) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_inpaint_pndm (line 476) | def test_inpaint_pndm(self): method test_inpaint_ddim (line 491) | def test_inpaint_ddim(self): method test_inpaint_lms (line 507) | def test_inpaint_lms(self): method test_inpaint_dpm (line 523) | def test_inpaint_dpm(self): FILE: tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py class StableDiffusionInstructPix2PixPipelineFastTests (line 44) | class StableDiffusionInstructPix2PixPipelineFastTests(PipelineTesterMixi... method get_dummy_components (line 49) | def get_dummy_components(self): method get_dummy_inputs (line 97) | def get_dummy_inputs(self, device, seed=0): method test_stable_diffusion_pix2pix_default_case (line 116) | def test_stable_diffusion_pix2pix_default_case(self): method test_stable_diffusion_pix2pix_negative_prompt (line 131) | def test_stable_diffusion_pix2pix_negative_prompt(self): method test_stable_diffusion_pix2pix_multiple_init_images (line 149) | def test_stable_diffusion_pix2pix_multiple_init_images(self): method test_stable_diffusion_pix2pix_euler (line 172) | def test_stable_diffusion_pix2pix_euler(self): class StableDiffusionInstructPix2PixPipelineSlowTests (line 197) | class StableDiffusionInstructPix2PixPipelineSlowTests(unittest.TestCase): method tearDown (line 198) | def tearDown(self): method get_inputs (line 203) | def get_inputs(self, seed=0): method test_stable_diffusion_pix2pix_default (line 219) | def test_stable_diffusion_pix2pix_default(self): method test_stable_diffusion_pix2pix_k_lms (line 236) | def test_stable_diffusion_pix2pix_k_lms(self): method test_stable_diffusion_pix2pix_ddim (line 254) | def test_stable_diffusion_pix2pix_ddim(self): method test_stable_diffusion_pix2pix_intermediate_state (line 272) | def test_stable_diffusion_pix2pix_intermediate_state(self): method test_stable_diffusion_pipeline_with_sequential_cpu_offloading (line 308) | def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): method test_stable_diffusion_pix2pix_pipeline_multiple_of_8 (line 328) | def test_stable_diffusion_pix2pix_pipeline_multiple_of_8(self): FILE: tests/pipelines/stable_diffusion/test_stable_diffusion_k_diffusion.py class StableDiffusionPipelineIntegrationTests (line 32) | class StableDiffusionPipelineIntegrationTests(unittest.TestCase): method tearDown (line 33) | def tearDown(self): method test_stable_diffusion_1 (line 39) | def test_stable_diffusion_1(self): method test_stable_diffusion_2 (line 59) | def test_stable_diffusion_2(self): FILE: tests/pipelines/stable_diffusion/test_stable_diffusion_panorama.py class StableDiffusionPanoramaPipelineFastTests (line 43) | class StableDiffusionPanoramaPipelineFastTests(PipelineTesterMixin, unit... method get_dummy_components (line 48) | def get_dummy_components(self): method get_dummy_inputs (line 96) | def get_dummy_inputs(self, device, seed=0): method test_stable_diffusion_panorama_default_case (line 110) | def test_stable_diffusion_panorama_default_case(self): method test_stable_diffusion_panorama_negative_prompt (line 126) | def test_stable_diffusion_panorama_negative_prompt(self): method test_stable_diffusion_panorama_euler (line 145) | def test_stable_diffusion_panorama_euler(self): method test_stable_diffusion_panorama_pndm (line 167) | def test_stable_diffusion_panorama_pndm(self): class StableDiffusionPanoramaSlowTests (line 183) | class StableDiffusionPanoramaSlowTests(unittest.TestCase): method tearDown (line 184) | def tearDown(self): method get_inputs (line 189) | def get_inputs(self, seed=0): method test_stable_diffusion_panorama_default (line 200) | def test_stable_diffusion_panorama_default(self): method test_stable_diffusion_panorama_k_lms (line 230) | def test_stable_diffusion_panorama_k_lms(self): method test_stable_diffusion_panorama_intermediate_state (line 263) | def test_stable_diffusion_panorama_intermediate_state(self): method test_stable_diffusion_panorama_pipeline_with_sequential_cpu_offloading (line 324) | def test_stable_diffusion_panorama_pipeline_with_sequential_cpu_offloa... FILE: tests/pipelines/stable_diffusion/test_stable_diffusion_pix2pix_zero.py class StableDiffusionPix2PixZeroPipelineFastTests (line 44) | class StableDiffusionPix2PixZeroPipelineFastTests(PipelineTesterMixin, u... method setUpClass (line 50) | def setUpClass(cls): method get_dummy_components (line 59) | def get_dummy_components(self): method get_dummy_inputs (line 110) | def get_dummy_inputs(self, device, seed=0): method test_stable_diffusion_pix2pix_zero_default_case (line 125) | def test_stable_diffusion_pix2pix_zero_default_case(self): method test_stable_diffusion_pix2pix_zero_negative_prompt (line 140) | def test_stable_diffusion_pix2pix_zero_negative_prompt(self): method test_stable_diffusion_pix2pix_zero_euler (line 158) | def test_stable_diffusion_pix2pix_zero_euler(self): method test_stable_diffusion_pix2pix_zero_ddpm (line 177) | def test_stable_diffusion_pix2pix_zero_ddpm(self): method test_inference_batch_single_identical (line 196) | def test_inference_batch_single_identical(self): class StableDiffusionPix2PixZeroPipelineSlowTests (line 202) | class StableDiffusionPix2PixZeroPipelineSlowTests(unittest.TestCase): method tearDown (line 203) | def tearDown(self): method setUpClass (line 209) | def setUpClass(cls): method get_inputs (line 218) | def get_inputs(self, seed=0): method test_stable_diffusion_pix2pix_zero_default (line 233) | def test_stable_diffusion_pix2pix_zero_default(self): method test_stable_diffusion_pix2pix_zero_k_lms (line 251) | def test_stable_diffusion_pix2pix_zero_k_lms(self): method test_stable_diffusion_pix2pix_zero_intermediate_state (line 269) | def test_stable_diffusion_pix2pix_zero_intermediate_state(self): method test_stable_diffusion_pipeline_with_sequential_cpu_offloading (line 306) | def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): class InversionPipelineSlowTests (line 330) | class InversionPipelineSlowTests(unittest.TestCase): method tearDown (line 331) | def tearDown(self): method setUpClass (line 337) | def setUpClass(cls): method test_stable_diffusion_pix2pix_inversion (line 346) | def test_stable_diffusion_pix2pix_inversion(self): method test_stable_diffusion_pix2pix_full (line 369) | def test_stable_diffusion_pix2pix_full(self): FILE: tests/pipelines/stable_diffusion/test_stable_diffusion_sag.py class StableDiffusionSAGPipelineFastTests (line 39) | class StableDiffusionSAGPipelineFastTests(PipelineTesterMixin, unittest.... method get_dummy_components (line 45) | def get_dummy_components(self): method get_dummy_inputs (line 99) | def get_dummy_inputs(self, device, seed=0): class StableDiffusionPipelineIntegrationTests (line 117) | class StableDiffusionPipelineIntegrationTests(unittest.TestCase): method tearDown (line 118) | def tearDown(self): method test_stable_diffusion_1 (line 124) | def test_stable_diffusion_1(self): method test_stable_diffusion_2 (line 144) | def test_stable_diffusion_2(self): FILE: tests/pipelines/stable_diffusion_2/test_stable_diffusion.py class StableDiffusion2PipelineFastTests (line 45) | class StableDiffusion2PipelineFastTests(PipelineTesterMixin, unittest.Te... method get_dummy_components (line 50) | def get_dummy_components(self): method get_dummy_inputs (line 111) | def get_dummy_inputs(self, device, seed=0): method test_stable_diffusion_ddim (line 125) | def test_stable_diffusion_ddim(self): method test_stable_diffusion_pndm (line 141) | def test_stable_diffusion_pndm(self): method test_stable_diffusion_k_lms (line 158) | def test_stable_diffusion_k_lms(self): method test_stable_diffusion_k_euler_ancestral (line 175) | def test_stable_diffusion_k_euler_ancestral(self): method test_stable_diffusion_k_euler (line 192) | def test_stable_diffusion_k_euler(self): method test_stable_diffusion_long_prompt (line 209) | def test_stable_diffusion_long_prompt(self): class StableDiffusion2PipelineSlowTests (line 250) | class StableDiffusion2PipelineSlowTests(unittest.TestCase): method tearDown (line 251) | def tearDown(self): method get_inputs (line 256) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_stable_diffusion_default_ddim (line 270) | def test_stable_diffusion_default_ddim(self): method test_stable_diffusion_pndm (line 283) | def test_stable_diffusion_pndm(self): method test_stable_diffusion_k_lms (line 297) | def test_stable_diffusion_k_lms(self): method test_stable_diffusion_attention_slicing (line 311) | def test_stable_diffusion_attention_slicing(self): method test_stable_diffusion_text2img_intermediate_state (line 339) | def test_stable_diffusion_text2img_intermediate_state(self): method test_stable_diffusion_pipeline_with_sequential_cpu_offloading (line 379) | def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): method test_stable_diffusion_pipeline_with_model_offloading (line 399) | def test_stable_diffusion_pipeline_with_model_offloading(self): class StableDiffusion2PipelineNightlyTests (line 453) | class StableDiffusion2PipelineNightlyTests(unittest.TestCase): method tearDown (line 454) | def tearDown(self): method get_inputs (line 459) | def get_inputs(self, device, generator_device="cpu", dtype=torch.float... method test_stable_diffusion_2_0_default_ddim (line 473) | def test_stable_diffusion_2_0_default_ddim(self): method test_stable_diffusion_2_1_default_pndm (line 487) | def test_stable_diffusion_2_1_default_pndm(self): method test_stable_diffusion_ddim (line 501) | def test_stable_diffusion_ddim(self): method test_stable_diffusion_lms (line 516) | def test_stable_diffusion_lms(self): method test_stable_diffusion_euler (line 531) | def test_stable_diffusion_euler(self): method test_stable_diffusion_dpm (line 546) | def test_stable_diffusion_dpm(self): FILE: tests/pipelines/stable_diffusion_2/test_stable_diffusion_attend_and_excite.py class StableDiffusionAttendAndExcitePipelineFastTests (line 37) | class StableDiffusionAttendAndExcitePipelineFastTests(PipelineTesterMixi... method get_dummy_components (line 43) | def get_dummy_components(self): method get_dummy_inputs (line 105) | def get_dummy_inputs(self, device, seed=0): method test_inference (line 122) | def test_inference(self): method test_inference_batch_consistent (line 141) | def test_inference_batch_consistent(self): class StableDiffusionAttendAndExcitePipelineIntegrationTests (line 148) | class StableDiffusionAttendAndExcitePipelineIntegrationTests(unittest.Te... method tearDown (line 149) | def tearDown(self): method test_attend_and_excite_fp16 (line 154) | def test_attend_and_excite_fp16(self): FILE: tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py class StableDiffusionDepth2ImgPipelineFastTests (line 62) | class StableDiffusionDepth2ImgPipelineFastTests(PipelineTesterMixin, uni... method get_dummy_components (line 69) | def get_dummy_components(self): method get_dummy_inputs (line 151) | def get_dummy_inputs(self, device, seed=0): method test_save_load_local (line 169) | def test_save_load_local(self): method test_save_load_float16 (line 191) | def test_save_load_float16(self): method test_float16_inference (line 223) | def test_float16_inference(self): method test_cpu_offload_forward_pass (line 246) | def test_cpu_offload_forward_pass(self): method test_dict_tuple_outputs_equivalent (line 262) | def test_dict_tuple_outputs_equivalent(self): method test_progress_bar (line 278) | def test_progress_bar(self): method test_stable_diffusion_depth2img_default_case (line 281) | def test_stable_diffusion_depth2img_default_case(self): method test_stable_diffusion_depth2img_negative_prompt (line 300) | def test_stable_diffusion_depth2img_negative_prompt(self): method test_stable_diffusion_depth2img_multiple_init_images (line 321) | def test_stable_diffusion_depth2img_multiple_init_images(self): method test_stable_diffusion_depth2img_pil (line 343) | def test_stable_diffusion_depth2img_pil(self): method test_attention_slicing_forward_pass (line 363) | def test_attention_slicing_forward_pass(self): class StableDiffusionDepth2ImgPipelineSlowTests (line 369) | class StableDiffusionDepth2ImgPipelineSlowTests(unittest.TestCase): method tearDown (line 370) | def tearDown(self): method get_inputs (line 375) | def get_inputs(self, device="cpu", dtype=torch.float32, seed=0): method test_stable_diffusion_depth2img_pipeline_default (line 391) | def test_stable_diffusion_depth2img_pipeline_default(self): method test_stable_diffusion_depth2img_pipeline_k_lms (line 408) | def test_stable_diffusion_depth2img_pipeline_k_lms(self): method test_stable_diffusion_depth2img_pipeline_ddim (line 426) | def test_stable_diffusion_depth2img_pipeline_ddim(self): method test_stable_diffusion_depth2img_intermediate_state (line 444) | def test_stable_diffusion_depth2img_intermediate_state(self): method test_stable_diffusion_pipeline_with_sequential_cpu_offloading (line 484) | def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): class StableDiffusionImg2ImgPipelineNightlyTests (line 507) | class StableDiffusionImg2ImgPipelineNightlyTests(unittest.TestCase): method tearDown (line 508) | def tearDown(self): method get_inputs (line 513) | def get_inputs(self, device="cpu", dtype=torch.float32, seed=0): method test_depth2img_pndm (line 529) | def test_depth2img_pndm(self): method test_depth2img_ddim (line 544) | def test_depth2img_ddim(self): method test_img2img_lms (line 560) | def test_img2img_lms(self): method test_img2img_dpm (line 576) | def test_img2img_dpm(self): FILE: tests/pipelines/stable_diffusion_2/test_stable_diffusion_flax.py class FlaxStableDiffusion2PipelineIntegrationTests (line 33) | class FlaxStableDiffusion2PipelineIntegrationTests(unittest.TestCase): method tearDown (line 34) | def tearDown(self): method test_stable_diffusion_flax (line 39) | def test_stable_diffusion_flax(self): method test_stable_diffusion_dpm_flax (line 68) | def test_stable_diffusion_dpm_flax(self): FILE: tests/pipelines/stable_diffusion_2/test_stable_diffusion_flax_inpaint.py class FlaxStableDiffusionInpaintPipelineIntegrationTests (line 33) | class FlaxStableDiffusionInpaintPipelineIntegrationTests(unittest.TestCa... method tearDown (line 34) | def tearDown(self): method test_stable_diffusion_inpaint_pipeline (line 39) | def test_stable_diffusion_inpaint_pipeline(self): FILE: tests/pipelines/stable_diffusion_2/test_stable_diffusion_inpaint.py class StableDiffusion2InpaintPipelineFastTests (line 36) | class StableDiffusion2InpaintPipelineFastTests(PipelineTesterMixin, unit... method get_dummy_components (line 41) | def get_dummy_components(self): method get_dummy_inputs (line 96) | def get_dummy_inputs(self, device, seed=0): method test_stable_diffusion_inpaint (line 117) | def test_stable_diffusion_inpaint(self): class StableDiffusionInpaintPipelineIntegrationTests (line 136) | class StableDiffusionInpaintPipelineIntegrationTests(unittest.TestCase): method tearDown (line 137) | def tearDown(self): method test_stable_diffusion_inpaint_pipeline (line 143) | def test_stable_diffusion_inpaint_pipeline(self): method test_stable_diffusion_inpaint_pipeline_fp16 (line 177) | def test_stable_diffusion_inpaint_pipeline_fp16(self): method test_stable_diffusion_pipeline_with_sequential_cpu_offloading (line 215) | def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): FILE: tests/pipelines/stable_diffusion_2/test_stable_diffusion_latent_upscale.py class StableDiffusionLatentUpscalePipelineFastTests (line 41) | class StableDiffusionLatentUpscalePipelineFastTests(PipelineTesterMixin,... method dummy_image (line 55) | def dummy_image(self): method get_dummy_components (line 63) | def get_dummy_components(self): method get_dummy_inputs (line 129) | def get_dummy_inputs(self, device, seed=0): method test_inference (line 143) | def test_inference(self): method test_inference_batch_single_identical (line 162) | def test_inference_batch_single_identical(self): class StableDiffusionLatentUpscalePipelineIntegrationTests (line 168) | class StableDiffusionLatentUpscalePipelineIntegrationTests(unittest.Test... method tearDown (line 169) | def tearDown(self): method test_latent_upscaler_fp16 (line 174) | def test_latent_upscaler_fp16(self): method test_latent_upscaler_fp16_image (line 203) | def test_latent_upscaler_fp16_image(self): FILE: tests/pipelines/stable_diffusion_2/test_stable_diffusion_upscale.py class StableDiffusionUpscalePipelineFastTests (line 33) | class StableDiffusionUpscalePipelineFastTests(unittest.TestCase): method tearDown (line 34) | def tearDown(self): method dummy_image (line 41) | def dummy_image(self): method dummy_cond_unet_upscale (line 50) | def dummy_cond_unet_upscale(self): method dummy_vae (line 70) | def dummy_vae(self): method dummy_text_encoder (line 83) | def dummy_text_encoder(self): method test_stable_diffusion_upscale (line 101) | def test_stable_diffusion_upscale(self): method test_stable_diffusion_upscale_batch (line 162) | def test_stable_diffusion_upscale_batch(self): method test_stable_diffusion_upscale_fp16 (line 214) | def test_stable_diffusion_upscale_fp16(self): class StableDiffusionUpscalePipelineIntegrationTests (line 259) | class StableDiffusionUpscalePipelineIntegrationTests(unittest.TestCase): method tearDown (line 260) | def tearDown(self): method test_stable_diffusion_upscale_pipeline (line 266) | def test_stable_diffusion_upscale_pipeline(self): method test_stable_diffusion_upscale_pipeline_fp16 (line 296) | def test_stable_diffusion_upscale_pipeline_fp16(self): method test_stable_diffusion_pipeline_with_sequential_cpu_offloading (line 329) | def test_stable_diffusion_pipeline_with_sequential_cpu_offloading(self): FILE: tests/pipelines/stable_diffusion_2/test_stable_diffusion_v_pred.py class StableDiffusion2VPredictionPipelineFastTests (line 39) | class StableDiffusion2VPredictionPipelineFastTests(unittest.TestCase): method tearDown (line 40) | def tearDown(self): method dummy_cond_unet (line 47) | def dummy_cond_unet(self): method dummy_vae (line 65) | def dummy_vae(self): method dummy_text_encoder (line 79) | def dummy_text_encoder(self): method test_stable_diffusion_v_pred_ddim (line 97) | def test_stable_diffusion_v_pred_ddim(self): method test_stable_diffusion_v_pred_k_euler (line 152) | def test_stable_diffusion_v_pred_k_euler(self): method test_stable_diffusion_v_pred_fp16 (line 202) | def test_stable_diffusion_v_pred_fp16(self): class StableDiffusion2VPredictionPipelineIntegrationTests (line 245) | class StableDiffusion2VPredictionPipelineIntegrationTests(unittest.TestC... method tearDown (line 246) | def tearDown(self): method test_stable_diffusion_v_pred_default (line 252) | def test_stable_diffusion_v_pred_default(self): method test_stable_diffusion_v_pred_upcast_attention (line 270) | def test_stable_diffusion_v_pred_upcast_attention(self): method test_stable_diffusion_v_pred_euler (line 290) | def test_stable_diffusion_v_pred_euler(self): method test_stable_diffusion_v_pred_dpm (line 310) | def test_stable_diffusion_v_pred_dpm(self): method test_stable_diffusion_attention_slicing_v_pred (line 334) | def test_stable_diffusion_attention_slicing_v_pred(self): method test_stable_diffusion_text2img_pipeline_v_pred_default (line 367) | def test_stable_diffusion_text2img_pipeline_v_pred_default(self): method test_stable_diffusion_text2img_pipeline_v_pred_fp16 (line 387) | def test_stable_diffusion_text2img_pipeline_v_pred_fp16(self): method test_stable_diffusion_text2img_intermediate_state_v_pred (line 406) | def test_stable_diffusion_text2img_intermediate_state_v_pred(self): method test_stable_diffusion_low_cpu_mem_usage_v_pred (line 449) | def test_stable_diffusion_low_cpu_mem_usage_v_pred(self): method test_stable_diffusion_pipeline_with_sequential_cpu_offloading_v_pred (line 463) | def test_stable_diffusion_pipeline_with_sequential_cpu_offloading_v_pr... FILE: tests/pipelines/stable_diffusion_safe/test_safe_diffusion.py class SafeDiffusionPipelineFastTests (line 34) | class SafeDiffusionPipelineFastTests(unittest.TestCase): method tearDown (line 35) | def tearDown(self): method dummy_image (line 42) | def dummy_image(self): method dummy_cond_unet (line 51) | def dummy_cond_unet(self): method dummy_vae (line 66) | def dummy_vae(self): method dummy_text_encoder (line 79) | def dummy_text_encoder(self): method dummy_extractor (line 95) | def dummy_extractor(self): method test_safe_diffusion_ddim (line 109) | def test_safe_diffusion_ddim(self): method test_stable_diffusion_pndm (line 162) | def test_stable_diffusion_pndm(self): method test_stable_diffusion_no_safety_checker (line 208) | def test_stable_diffusion_no_safety_checker(self): method test_stable_diffusion_fp16 (line 230) | def test_stable_diffusion_fp16(self): class SafeDiffusionPipelineIntegrationTests (line 264) | class SafeDiffusionPipelineIntegrationTests(unittest.TestCase): method tearDown (line 265) | def tearDown(self): method test_harm_safe_stable_diffusion (line 271) | def test_harm_safe_stable_diffusion(self): method test_nudity_safe_stable_diffusion (line 332) | def test_nudity_safe_stable_diffusion(self): method test_nudity_safetychecker_safe_stable_diffusion (line 386) | def test_nudity_safetychecker_safe_stable_diffusion(self): FILE: tests/pipelines/stable_unclip/test_stable_unclip.py class StableUnCLIPPipelineFastTests (line 22) | class StableUnCLIPPipelineFastTests(PipelineTesterMixin, unittest.TestCa... method get_dummy_components (line 30) | def get_dummy_components(self): method get_dummy_inputs (line 149) | def get_dummy_inputs(self, device, seed=0): method test_attention_slicing_forward_pass (line 165) | def test_attention_slicing_forward_pass(self): method test_inference_batch_single_identical (line 172) | def test_inference_batch_single_identical(self): class StableUnCLIPPipelineIntegrationTests (line 180) | class StableUnCLIPPipelineIntegrationTests(unittest.TestCase): method tearDown (line 181) | def tearDown(self): method test_stable_unclip (line 187) | def test_stable_unclip(self): method test_stable_unclip_pipeline_with_sequential_cpu_offloading (line 209) | def test_stable_unclip_pipeline_with_sequential_cpu_offloading(self): FILE: tests/pipelines/stable_unclip/test_stable_unclip_img2img.py class StableUnCLIPImg2ImgPipelineFastTests (line 28) | class StableUnCLIPImg2ImgPipelineFastTests(PipelineTesterMixin, unittest... method get_dummy_components (line 33) | def get_dummy_components(self): method get_dummy_inputs (line 127) | def get_dummy_inputs(self, device, seed=0, pil_image=True): method test_attention_slicing_forward_pass (line 151) | def test_attention_slicing_forward_pass(self): method test_inference_batch_single_identical (line 158) | def test_inference_batch_single_identical(self): method test_xformers_attention_forwardGenerator_pass (line 167) | def test_xformers_attention_forwardGenerator_pass(self): class StableUnCLIPImg2ImgPipelineIntegrationTests (line 173) | class StableUnCLIPImg2ImgPipelineIntegrationTests(unittest.TestCase): method tearDown (line 174) | def tearDown(self): method test_stable_unclip_l_img2img (line 180) | def test_stable_unclip_l_img2img(self): method test_stable_unclip_h_img2img (line 208) | def test_stable_unclip_h_img2img(self): method test_stable_unclip_img2img_pipeline_with_sequential_cpu_offloading (line 236) | def test_stable_unclip_img2img_pipeline_with_sequential_cpu_offloading... FILE: tests/pipelines/test_pipeline_utils.py class IsSafetensorsCompatibleTests (line 6) | class IsSafetensorsCompatibleTests(unittest.TestCase): method test_all_is_compatible (line 7) | def test_all_is_compatible(self): method test_diffusers_model_is_compatible (line 20) | def test_diffusers_model_is_compatible(self): method test_diffusers_model_is_not_compatible (line 27) | def test_diffusers_model_is_not_compatible(self): method test_transformer_model_is_compatible (line 40) | def test_transformer_model_is_compatible(self): method test_transformer_model_is_not_compatible (line 47) | def test_transformer_model_is_not_compatible(self): method test_all_is_compatible_variant (line 60) | def test_all_is_compatible_variant(self): method test_diffusers_model_is_compatible_variant (line 74) | def test_diffusers_model_is_compatible_variant(self): method test_diffusers_model_is_compatible_variant_partial (line 82) | def test_diffusers_model_is_compatible_variant_partial(self): method test_diffusers_model_is_not_compatible_variant (line 91) | def test_diffusers_model_is_not_compatible_variant(self): method test_transformer_model_is_compatible_variant (line 105) | def test_transformer_model_is_compatible_variant(self): method test_transformer_model_is_compatible_variant_partial (line 113) | def test_transformer_model_is_compatible_variant_partial(self): method test_transformer_model_is_not_compatible_variant (line 122) | def test_transformer_model_is_not_compatible_variant(self): FILE: tests/pipelines/unclip/test_unclip.py class UnCLIPPipelineFastTests (line 32) | class UnCLIPPipelineFastTests(PipelineTesterMixin, unittest.TestCase): method text_embedder_hidden_size (line 54) | def text_embedder_hidden_size(self): method time_input_dim (line 58) | def time_input_dim(self): method block_out_channels_0 (line 62) | def block_out_channels_0(self): method time_embed_dim (line 66) | def time_embed_dim(self): method cross_attention_dim (line 70) | def cross_attention_dim(self): method dummy_tokenizer (line 74) | def dummy_tokenizer(self): method dummy_text_encoder (line 79) | def dummy_text_encoder(self): method dummy_prior (line 96) | def dummy_prior(self): method dummy_text_proj (line 110) | def dummy_text_proj(self): method dummy_decoder (line 123) | def dummy_decoder(self): method dummy_super_res_kwargs (line 147) | def dummy_super_res_kwargs(self): method dummy_super_res_first (line 159) | def dummy_super_res_first(self): method dummy_super_res_last (line 166) | def dummy_super_res_last(self): method get_dummy_components (line 173) | def get_dummy_components(self): method get_dummy_inputs (line 216) | def get_dummy_inputs(self, device, seed=0): method test_unclip (line 231) | def test_unclip(self): method test_unclip_passed_text_embed (line 271) | def test_unclip_passed_text_embed(self): method test_attention_slicing_forward_pass (line 358) | def test_attention_slicing_forward_pass(self): method test_inference_batch_single_identical (line 366) | def test_inference_batch_single_identical(self): method test_inference_batch_consistent (line 381) | def test_inference_batch_consistent(self): method test_dict_tuple_outputs_equivalent (line 401) | def test_dict_tuple_outputs_equivalent(self): method test_save_load_local (line 405) | def test_save_load_local(self): method test_save_load_optional_components (line 409) | def test_save_load_optional_components(self): class UnCLIPPipelineCPUIntegrationTests (line 414) | class UnCLIPPipelineCPUIntegrationTests(unittest.TestCase): method tearDown (line 415) | def tearDown(self): method test_unclip_karlo_cpu_fp32 (line 421) | def test_unclip_karlo_cpu_fp32(self): class UnCLIPPipelineIntegrationTests (line 446) | class UnCLIPPipelineIntegrationTests(unittest.TestCase): method tearDown (line 447) | def tearDown(self): method test_unclip_karlo (line 453) | def test_unclip_karlo(self): method test_unclip_pipeline_with_sequential_cpu_offloading (line 476) | def test_unclip_pipeline_with_sequential_cpu_offloading(self): FILE: tests/pipelines/unclip/test_unclip_image_variation.py class UnCLIPImageVariationPipelineFastTests (line 46) | class UnCLIPImageVariationPipelineFastTests(PipelineTesterMixin, unittes... method text_embedder_hidden_size (line 59) | def text_embedder_hidden_size(self): method time_input_dim (line 63) | def time_input_dim(self): method block_out_channels_0 (line 67) | def block_out_channels_0(self): method time_embed_dim (line 71) | def time_embed_dim(self): method cross_attention_dim (line 75) | def cross_attention_dim(self): method dummy_tokenizer (line 79) | def dummy_tokenizer(self): method dummy_text_encoder (line 84) | def dummy_text_encoder(self): method dummy_image_encoder (line 101) | def dummy_image_encoder(self): method dummy_text_proj (line 115) | def dummy_text_proj(self): method dummy_decoder (line 128) | def dummy_decoder(self): method dummy_super_res_kwargs (line 152) | def dummy_super_res_kwargs(self): method dummy_super_res_first (line 164) | def dummy_super_res_first(self): method dummy_super_res_last (line 171) | def dummy_super_res_last(self): method get_dummy_components (line 178) | def get_dummy_components(self): method get_dummy_inputs (line 215) | def get_dummy_inputs(self, device, seed=0, pil_image=True): method test_unclip_image_variation_input_tensor (line 236) | def test_unclip_image_variation_input_tensor(self): method test_unclip_image_variation_input_image (line 280) | def test_unclip_image_variation_input_image(self): method test_unclip_image_variation_input_list_images (line 312) | def test_unclip_image_variation_input_list_images(self): method test_unclip_passed_image_embed (line 364) | def test_unclip_passed_image_embed(self): method test_attention_slicing_forward_pass (line 420) | def test_attention_slicing_forward_pass(self): method test_inference_batch_single_identical (line 428) | def test_inference_batch_single_identical(self): method test_inference_batch_consistent (line 442) | def test_inference_batch_consistent(self): method test_dict_tuple_outputs_equivalent (line 461) | def test_dict_tuple_outputs_equivalent(self): method test_save_load_local (line 465) | def test_save_load_local(self): method test_save_load_optional_components (line 469) | def test_save_load_optional_components(self): class UnCLIPImageVariationPipelineIntegrationTests (line 475) | class UnCLIPImageVariationPipelineIntegrationTests(unittest.TestCase): method tearDown (line 476) | def tearDown(self): method test_unclip_image_variation_karlo (line 482) | def test_unclip_image_variation_karlo(self): FILE: tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py class VersatileDiffusionDualGuidedPipelineFastTests (line 30) | class VersatileDiffusionDualGuidedPipelineFastTests(unittest.TestCase): class VersatileDiffusionDualGuidedPipelineIntegrationTests (line 36) | class VersatileDiffusionDualGuidedPipelineIntegrationTests(unittest.Test... method tearDown (line 37) | def tearDown(self): method test_remove_unused_weights_save_load (line 43) | def test_remove_unused_weights_save_load(self): method test_inference_dual_guided (line 85) | def test_inference_dual_guided(self): FILE: tests/pipelines/versatile_diffusion/test_versatile_diffusion_image_variation.py class VersatileDiffusionImageVariationPipelineFastTests (line 28) | class VersatileDiffusionImageVariationPipelineFastTests(unittest.TestCase): class VersatileDiffusionImageVariationPipelineIntegrationTests (line 34) | class VersatileDiffusionImageVariationPipelineIntegrationTests(unittest.... method test_inference_image_variations (line 35) | def test_inference_image_variations(self): FILE: tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py class VersatileDiffusionMegaPipelineFastTests (line 30) | class VersatileDiffusionMegaPipelineFastTests(unittest.TestCase): class VersatileDiffusionMegaPipelineIntegrationTests (line 36) | class VersatileDiffusionMegaPipelineIntegrationTests(unittest.TestCase): method tearDown (line 37) | def tearDown(self): method test_from_save_pretrained (line 43) | def test_from_save_pretrained(self): method test_inference_dual_guided_then_text_to_image (line 82) | def test_inference_dual_guided_then_text_to_image(self): FILE: tests/pipelines/versatile_diffusion/test_versatile_diffusion_text_to_image.py class VersatileDiffusionTextToImagePipelineFastTests (line 30) | class VersatileDiffusionTextToImagePipelineFastTests(unittest.TestCase): class VersatileDiffusionTextToImagePipelineIntegrationTests (line 36) | class VersatileDiffusionTextToImagePipelineIntegrationTests(unittest.Tes... method tearDown (line 37) | def tearDown(self): method test_remove_unused_weights_save_load (line 43) | def test_remove_unused_weights_save_load(self): method test_inference_text2img (line 69) | def test_inference_text2img(self): FILE: tests/pipelines/vq_diffusion/test_vq_diffusion.py class VQDiffusionPipelineFastTests (line 32) | class VQDiffusionPipelineFastTests(unittest.TestCase): method tearDown (line 33) | def tearDown(self): method num_embed (line 40) | def num_embed(self): method num_embeds_ada_norm (line 44) | def num_embeds_ada_norm(self): method text_embedder_hidden_size (line 48) | def text_embedder_hidden_size(self): method dummy_vqvae (line 52) | def dummy_vqvae(self): method dummy_tokenizer (line 67) | def dummy_tokenizer(self): method dummy_text_encoder (line 72) | def dummy_text_encoder(self): method dummy_transformer (line 88) | def dummy_transformer(self): method test_vq_diffusion (line 109) | def test_vq_diffusion(self): method test_vq_diffusion_classifier_free_sampling (line 151) | def test_vq_diffusion_classifier_free_sampling(self): class VQDiffusionPipelineIntegrationTests (line 198) | class VQDiffusionPipelineIntegrationTests(unittest.TestCase): method tearDown (line 199) | def tearDown(self): method test_vq_diffusion_classifier_free_sampling (line 205) | def test_vq_diffusion_classifier_free_sampling(self): FILE: tests/repo_utils/test_check_copies.py class CopyCheckTester (line 50) | class CopyCheckTester(unittest.TestCase): method setUp (line 51) | def setUp(self): method tearDown (line 60) | def tearDown(self): method check_copy_consistency (line 64) | def check_copy_consistency(self, comment, class_name, class_code, over... method test_find_code_in_diffusers (line 80) | def test_find_code_in_diffusers(self): method test_is_copy_consistent (line 84) | def test_is_copy_consistent(self): FILE: tests/repo_utils/test_check_dummies.py class CheckDummiesTester (line 31) | class CheckDummiesTester(unittest.TestCase): method test_find_backend (line 32) | def test_find_backend(self): method test_read_init (line 52) | def test_read_init(self): method test_create_dummy_object (line 68) | def test_create_dummy_object(self): method test_create_dummy_files (line 95) | def test_create_dummy_files(self): FILE: tests/test_config.py class SampleObject (line 32) | class SampleObject(ConfigMixin): method __init__ (line 36) | def __init__( class SampleObject2 (line 47) | class SampleObject2(ConfigMixin): method __init__ (line 51) | def __init__( class SampleObject3 (line 62) | class SampleObject3(ConfigMixin): method __init__ (line 66) | def __init__( class ConfigTester (line 78) | class ConfigTester(unittest.TestCase): method test_load_not_from_mixin (line 79) | def test_load_not_from_mixin(self): method test_register_to_config (line 83) | def test_register_to_config(self): method test_save_load (line 119) | def test_save_load(self): method test_load_ddim_from_pndm (line 142) | def test_load_ddim_from_pndm(self): method test_load_euler_from_pndm (line 154) | def test_load_euler_from_pndm(self): method test_load_euler_ancestral_from_pndm (line 166) | def test_load_euler_ancestral_from_pndm(self): method test_load_pndm (line 178) | def test_load_pndm(self): method test_overwrite_config_on_load (line 190) | def test_overwrite_config_on_load(self): method test_load_dpmsolver (line 213) | def test_load_dpmsolver(self): FILE: tests/test_hub_utils.py class CreateModelCardTest (line 23) | class CreateModelCardTest(unittest.TestCase): method test_create_model_card (line 25) | def test_create_model_card(self, repo_name_mock: Mock) -> None: FILE: tests/test_layers_utils.py class EmbeddingsTests (line 33) | class EmbeddingsTests(unittest.TestCase): method test_timestep_embeddings (line 34) | def test_timestep_embeddings(self): method test_timestep_defaults (line 57) | def test_timestep_defaults(self): method test_timestep_flip_sin_cos (line 68) | def test_timestep_flip_sin_cos(self): method test_timestep_downscale_freq_shift (line 79) | def test_timestep_downscale_freq_shift(self): method test_sinoid_embeddings_hardcoded (line 92) | def test_sinoid_embeddings_hardcoded(self): class Upsample2DBlockTests (line 120) | class Upsample2DBlockTests(unittest.TestCase): method test_upsample_default (line 121) | def test_upsample_default(self): method test_upsample_with_conv (line 133) | def test_upsample_with_conv(self): method test_upsample_with_conv_out_dim (line 145) | def test_upsample_with_conv_out_dim(self): method test_upsample_with_transpose (line 157) | def test_upsample_with_transpose(self): class Downsample2DBlockTests (line 170) | class Downsample2DBlockTests(unittest.TestCase): method test_downsample_default (line 171) | def test_downsample_default(self): method test_downsample_with_conv (line 185) | def test_downsample_with_conv(self): method test_downsample_with_conv_pad1 (line 200) | def test_downsample_with_conv_pad1(self): method test_downsample_with_conv_out_dim (line 212) | def test_downsample_with_conv_out_dim(self): class ResnetBlock2DTests (line 225) | class ResnetBlock2DTests(unittest.TestCase): method test_resnet_default (line 226) | def test_resnet_default(self): method test_restnet_with_use_in_shortcut (line 241) | def test_restnet_with_use_in_shortcut(self): method test_resnet_up (line 256) | def test_resnet_up(self): method test_resnet_down (line 271) | def test_resnet_down(self): method test_restnet_with_kernel_fir (line 286) | def test_restnet_with_kernel_fir(self): method test_restnet_with_kernel_sde_vp (line 301) | def test_restnet_with_kernel_sde_vp(self): class AttentionBlockTests (line 317) | class AttentionBlockTests(unittest.TestCase): method test_attention_block_default (line 321) | def test_attention_block_default(self): method test_attention_block_sd (line 345) | def test_attention_block_sd(self): class Transformer2DModelTests (line 370) | class Transformer2DModelTests(unittest.TestCase): method test_spatial_transformer_default (line 371) | def test_spatial_transformer_default(self): method test_spatial_transformer_cross_attention_dim (line 395) | def test_spatial_transformer_cross_attention_dim(self): method test_spatial_transformer_timestep (line 420) | def test_spatial_transformer_timestep(self): method test_spatial_transformer_dropout (line 458) | def test_spatial_transformer_dropout(self): method test_spatial_transformer_discrete (line 487) | def test_spatial_transformer_discrete(self): method test_spatial_transformer_default_norm_layers (line 516) | def test_spatial_transformer_default_norm_layers(self): method test_spatial_transformer_ada_norm_layers (line 522) | def test_spatial_transformer_ada_norm_layers(self): method test_spatial_transformer_default_ff_layers (line 533) | def test_spatial_transformer_default_ff_layers(self): method test_spatial_transformer_geglu_approx_ff_layers (line 556) | def test_spatial_transformer_geglu_approx_ff_layers(self): method test_spatial_transformer_attention_bias (line 579) | def test_spatial_transformer_attention_bias(self): FILE: tests/test_modeling_common.py class ModelUtilsTest (line 31) | class ModelUtilsTest(unittest.TestCase): method test_accelerate_loading_error_message (line 32) | def test_accelerate_loading_error_message(self): method test_cached_files_are_used_when_no_internet (line 39) | def test_cached_files_are_used_when_no_internet(self): class ModelTesterMixin (line 64) | class ModelTesterMixin: method test_from_save_pretrained (line 65) | def test_from_save_pretrained(self): method test_from_save_pretrained_variant (line 95) | def test_from_save_pretrained_variant(self): method test_from_save_pretrained_dtype (line 133) | def test_from_save_pretrained_dtype(self): method test_determinism (line 151) | def test_determinism(self): method test_output (line 177) | def test_output(self): method test_forward_with_norm_groups (line 193) | def test_forward_with_norm_groups(self): method test_forward_signature (line 213) | def test_forward_signature(self): method test_model_from_pretrained (line 224) | def test_model_from_pretrained(self): method test_training (line 259) | def test_training(self): method test_ema_training (line 275) | def test_ema_training(self): method test_outputs_equivalence (line 293) | def test_outputs_equivalence(self): method test_enable_disable_gradient_checkpointing (line 342) | def test_enable_disable_gradient_checkpointing(self): method test_deprecated_kwargs (line 360) | def test_deprecated_kwargs(self): FILE: tests/test_modeling_common_flax.py class FlaxModelTesterMixin (line 12) | class FlaxModelTesterMixin: method test_output (line 13) | def test_output(self): method test_forward_with_norm_groups (line 29) | def test_forward_with_norm_groups(self): method test_deprecated_kwargs (line 48) | def test_deprecated_kwargs(self): FILE: tests/test_outputs.py class CustomOutput (line 12) | class CustomOutput(BaseOutput): class ConfigTester (line 16) | class ConfigTester(unittest.TestCase): method test_outputs_single_attribute (line 17) | def test_outputs_single_attribute(self): method test_outputs_dict_init (line 39) | def test_outputs_dict_init(self): FILE: tests/test_pipelines.py class DownloadTests (line 62) | class DownloadTests(unittest.TestCase): method test_download_only_pytorch (line 63) | def test_download_only_pytorch(self): method test_returned_cached_folder (line 79) | def test_returned_cached_folder(self): method test_download_safetensors (line 100) | def test_download_safetensors(self): method test_download_no_safety_checker (line 116) | def test_download_no_safety_checker(self): method test_load_no_safety_checker_explicit_locally (line 132) | def test_load_no_safety_checker_explicit_locally(self): method test_load_no_safety_checker_default_locally (line 152) | def test_load_no_safety_checker_default_locally(self): method test_cached_files_are_used_when_no_internet (line 171) | def test_cached_files_are_used_when_no_internet(self): method test_download_from_variant_folder (line 198) | def test_download_from_variant_folder(self): method test_download_variant_all (line 223) | def test_download_variant_all(self): method test_download_variant_partly (line 253) | def test_download_variant_partly(self): method test_download_broken_variant (line 285) | def test_download_broken_variant(self): class CustomPipelineTests (line 321) | class CustomPipelineTests(unittest.TestCase): method test_load_custom_pipeline (line 322) | def test_load_custom_pipeline(self): method test_load_custom_github (line 331) | def test_load_custom_github(self): method test_run_custom_pipeline (line 356) | def test_run_custom_pipeline(self): method test_local_custom_pipeline_repo (line 368) | def test_local_custom_pipeline_repo(self): method test_local_custom_pipeline_file (line 381) | def test_local_custom_pipeline_file(self): method test_load_pipeline_from_git (line 397) | def test_load_pipeline_from_git(self): class PipelineFastTests (line 421) | class PipelineFastTests(unittest.TestCase): method tearDown (line 422) | def tearDown(self): method dummy_image (line 432) | def dummy_image(self): method dummy_uncond_unet (line 440) | def dummy_uncond_unet(self, sample_size=32): method dummy_cond_unet (line 453) | def dummy_cond_unet(self, sample_size=32): method dummy_vae (line 468) | def dummy_vae(self): method dummy_text_encoder (line 481) | def dummy_text_encoder(self): method dummy_extractor (line 497) | def dummy_extractor(self): method test_uncond_unet_components (line 519) | def test_uncond_unet_components(self, scheduler_fn=DDPMScheduler, pipe... method test_stable_diffusion_components (line 533) | def test_stable_diffusion_components(self): method test_pipe_false_offload_warn (line 588) | def test_pipe_false_offload_warn(self): method test_set_scheduler (line 623) | def test_set_scheduler(self): method test_set_scheduler_consistency (line 655) | def test_set_scheduler_consistency(self): method test_save_safe_serialization (line 699) | def test_save_safe_serialization(self): method test_no_pytorch_download_when_doing_safetensors (line 726) | def test_no_pytorch_download_when_doing_safetensors(self): method test_no_safetensors_download_when_doing_pytorch (line 745) | def test_no_safetensors_download_when_doing_pytorch(self): method test_optional_components (line 770) | def test_optional_components(self): class PipelineSlowTests (line 877) | class PipelineSlowTests(unittest.TestCase): method tearDown (line 878) | def tearDown(self): method test_smart_download (line 884) | def test_smart_download(self): method test_warning_unused_kwargs (line 905) | def test_warning_unused_kwargs(self): method test_from_save_pretrained (line 922) | def test_from_save_pretrained(self): method test_from_pretrained_hub (line 952) | def test_from_pretrained_hub(self): method test_from_pretrained_hub_pass_model (line 973) | def test_from_pretrained_hub_pass_model(self): method test_output_format (line 996) | def test_output_format(self): method test_from_flax_from_pt (line 1018) | def test_from_flax_from_pt(self): class PipelineNightlyTests (line 1064) | class PipelineNightlyTests(unittest.TestCase): method tearDown (line 1065) | def tearDown(self): method test_ddpm_ddim_equality_batched (line 1071) | def test_ddpm_ddim_equality_batched(self): FILE: tests/test_pipelines_common.py class PipelineTesterMixin (line 24) | class PipelineTesterMixin: method get_generator (line 52) | def get_generator(self, seed): method pipeline_class (line 58) | def pipeline_class(self) -> Union[Callable, DiffusionPipeline]: method get_dummy_components (line 64) | def get_dummy_components(self): method get_dummy_inputs (line 70) | def get_dummy_inputs(self, device, seed=0): method params (line 77) | def params(self) -> frozenset: method batch_params (line 92) | def batch_params(self) -> frozenset: method tearDown (line 105) | def tearDown(self): method test_save_load_local (line 111) | def test_save_load_local(self): method test_pipeline_call_signature (line 136) | def test_pipeline_call_signature(self): method test_inference_batch_consistent (line 175) | def test_inference_batch_consistent(self): method _test_inference_batch_consistent (line 178) | def _test_inference_batch_consistent( method test_inference_batch_single_identical (line 235) | def test_inference_batch_single_identical(self): method _test_inference_batch_single_identical (line 238) | def _test_inference_batch_single_identical( method test_dict_tuple_outputs_equivalent (line 317) | def test_dict_tuple_outputs_equivalent(self): method test_components_function (line 333) | def test_components_function(self): method test_float16_inference (line 341) | def test_float16_inference(self): method test_save_load_float16 (line 361) | def test_save_load_float16(self): method test_save_load_optional_components (line 392) | def test_save_load_optional_components(self): method test_to_device (line 431) | def test_to_device(self): method test_attention_slicing_forward_pass (line 450) | def test_attention_slicing_forward_pass(self): method _test_attention_slicing_forward_pass (line 453) | def _test_attention_slicing_forward_pass( method test_cpu_offload_forward_pass (line 486) | def test_cpu_offload_forward_pass(self): method test_xformers_attention_forwardGenerator_pass (line 509) | def test_xformers_attention_forwardGenerator_pass(self): method _test_xformers_attention_forwardGenerator_pass (line 512) | def _test_xformers_attention_forwardGenerator_pass(self, test_max_diff... method test_progress_bar (line 534) | def test_progress_bar(self): method test_num_images_per_prompt (line 556) | def test_num_images_per_prompt(self): function assert_mean_pixel_difference (line 586) | def assert_mean_pixel_difference(image, expected_image): FILE: tests/test_pipelines_flax.py class DownloadTests (line 37) | class DownloadTests(unittest.TestCase): method test_download_only_pytorch (line 38) | def test_download_only_pytorch(self): class FlaxPipelineTests (line 55) | class FlaxPipelineTests(unittest.TestCase): method test_dummy_all_tpus (line 56) | def test_dummy_all_tpus(self): method test_stable_diffusion_v1_4 (line 91) | def test_stable_diffusion_v1_4(self): method test_stable_diffusion_v1_4_bfloat_16 (line 122) | def test_stable_diffusion_v1_4_bfloat_16(self): method test_stable_diffusion_v1_4_bfloat_16_with_safety (line 153) | def test_stable_diffusion_v1_4_bfloat_16_with_safety(self): method test_stable_diffusion_v1_4_bfloat_16_ddim (line 182) | def test_stable_diffusion_v1_4_bfloat_16_ddim(self): FILE: tests/test_pipelines_onnx_common.py class OnnxPipelineTesterMixin (line 5) | class OnnxPipelineTesterMixin: FILE: tests/test_scheduler.py class SchedulerObject (line 56) | class SchedulerObject(SchedulerMixin, ConfigMixin): method __init__ (line 60) | def __init__( class SchedulerObject2 (line 71) | class SchedulerObject2(SchedulerMixin, ConfigMixin): method __init__ (line 75) | def __init__( class SchedulerObject3 (line 86) | class SchedulerObject3(SchedulerMixin, ConfigMixin): method __init__ (line 90) | def __init__( class SchedulerBaseTests (line 102) | class SchedulerBaseTests(unittest.TestCase): method test_save_load_from_different_config (line 103) | def test_save_load_from_different_config(self): method test_save_load_compatible_schedulers (line 145) | def test_save_load_compatible_schedulers(self): method test_save_load_from_different_config_comp_schedulers (line 181) | def test_save_load_from_different_config_comp_schedulers(self): class SchedulerCommonTest (line 219) | class SchedulerCommonTest(unittest.TestCase): method dummy_sample (line 224) | def dummy_sample(self): method dummy_sample_deter (line 235) | def dummy_sample_deter(self): method get_scheduler_config (line 249) | def get_scheduler_config(self): method dummy_model (line 252) | def dummy_model(self): method check_over_configs (line 258) | def check_over_configs(self, time_step=0, **config): method check_over_forward (line 306) | def check_over_forward(self, time_step=0, **forward_kwargs): method test_from_save_pretrained (line 348) | def test_from_save_pretrained(self): method test_compatibles (line 390) | def test_compatibles(self): method test_from_pretrained (line 414) | def test_from_pretrained(self): method test_step_shape (line 426) | def test_step_shape(self): method test_scheduler_outputs_equivalence (line 462) | def test_scheduler_outputs_equivalence(self): method test_scheduler_public_api (line 534) | def test_scheduler_public_api(self): method test_add_noise_device (line 561) | def test_add_noise_device(self): method test_deprecated_kwargs (line 578) | def test_deprecated_kwargs(self): method test_trained_betas (line 599) | def test_trained_betas(self): class DDPMSchedulerTest (line 614) | class DDPMSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 617) | def get_scheduler_config(self, **kwargs): method test_timesteps (line 630) | def test_timesteps(self): method test_betas (line 634) | def test_betas(self): method test_schedules (line 638) | def test_schedules(self): method test_variance_type (line 642) | def test_variance_type(self): method test_clip_sample (line 646) | def test_clip_sample(self): method test_prediction_type (line 650) | def test_prediction_type(self): method test_time_indices (line 654) | def test_time_indices(self): method test_variance (line 658) | def test_variance(self): method test_full_loop_no_noise (line 667) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 698) | def test_full_loop_with_v_prediction(self): class DDIMSchedulerTest (line 730) | class DDIMSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 734) | def get_scheduler_config(self, **kwargs): method full_loop (line 746) | def full_loop(self, **config): method test_timesteps (line 764) | def test_timesteps(self): method test_steps_offset (line 768) | def test_steps_offset(self): method test_betas (line 778) | def test_betas(self): method test_schedules (line 782) | def test_schedules(self): method test_prediction_type (line 786) | def test_prediction_type(self): method test_clip_sample (line 790) | def test_clip_sample(self): method test_time_indices (line 794) | def test_time_indices(self): method test_inference_steps (line 798) | def test_inference_steps(self): method test_eta (line 802) | def test_eta(self): method test_variance (line 806) | def test_variance(self): method test_full_loop_no_noise (line 818) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 827) | def test_full_loop_with_v_prediction(self): method test_full_loop_with_set_alpha_to_one (line 836) | def test_full_loop_with_set_alpha_to_one(self): method test_full_loop_with_no_set_alpha_to_one (line 845) | def test_full_loop_with_no_set_alpha_to_one(self): class DPMSolverSinglestepSchedulerTest (line 855) | class DPMSolverSinglestepSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 859) | def get_scheduler_config(self, **kwargs): method check_over_configs (line 876) | def check_over_configs(self, time_step=0, **config): method test_from_save_pretrained (line 904) | def test_from_save_pretrained(self): method check_over_forward (line 907) | def check_over_forward(self, time_step=0, **forward_kwargs): method full_loop (line 936) | def full_loop(self, **config): method test_timesteps (line 952) | def test_timesteps(self): method test_thresholding (line 956) | def test_thresholding(self): method test_prediction_type (line 971) | def test_prediction_type(self): method test_solver_order_and_type (line 975) | def test_solver_order_and_type(self): method test_lower_order_final (line 994) | def test_lower_order_final(self): method test_inference_steps (line 998) | def test_inference_steps(self): method test_full_loop_no_noise (line 1002) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 1008) | def test_full_loop_with_v_prediction(self): method test_fp16_support (line 1014) | def test_fp16_support(self): class DPMSolverMultistepSchedulerTest (line 1031) | class DPMSolverMultistepSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 1035) | def get_scheduler_config(self, **kwargs): method check_over_configs (line 1053) | def check_over_configs(self, time_step=0, **config): method test_from_save_pretrained (line 1081) | def test_from_save_pretrained(self): method check_over_forward (line 1084) | def check_over_forward(self, time_step=0, **forward_kwargs): method full_loop (line 1113) | def full_loop(self, **config): method test_step_shape (line 1129) | def test_step_shape(self): method test_timesteps (line 1159) | def test_timesteps(self): method test_thresholding (line 1163) | def test_thresholding(self): method test_prediction_type (line 1178) | def test_prediction_type(self): method test_solver_order_and_type (line 1182) | def test_solver_order_and_type(self): method test_lower_order_final (line 1201) | def test_lower_order_final(self): method test_inference_steps (line 1205) | def test_inference_steps(self): method test_full_loop_no_noise (line 1209) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 1215) | def test_full_loop_with_v_prediction(self): method test_fp16_support (line 1221) | def test_fp16_support(self): class PNDMSchedulerTest (line 1238) | class PNDMSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 1242) | def get_scheduler_config(self, **kwargs): method check_over_configs (line 1253) | def check_over_configs(self, time_step=0, **config): method test_from_save_pretrained (line 1284) | def test_from_save_pretrained(self): method check_over_forward (line 1287) | def check_over_forward(self, time_step=0, **forward_kwargs): method full_loop (line 1321) | def full_loop(self, **config): method test_step_shape (line 1341) | def test_step_shape(self): method test_timesteps (line 1374) | def test_timesteps(self): method test_steps_offset (line 1378) | def test_steps_offset(self): method test_betas (line 1393) | def test_betas(self): method test_schedules (line 1397) | def test_schedules(self): method test_prediction_type (line 1401) | def test_prediction_type(self): method test_time_indices (line 1405) | def test_time_indices(self): method test_inference_steps (line 1409) | def test_inference_steps(self): method test_pow_of_3_inference_steps (line 1413) | def test_pow_of_3_inference_steps(self): method test_inference_plms_no_past_residuals (line 1430) | def test_inference_plms_no_past_residuals(self): method test_full_loop_no_noise (line 1438) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 1446) | def test_full_loop_with_v_prediction(self): method test_full_loop_with_set_alpha_to_one (line 1454) | def test_full_loop_with_set_alpha_to_one(self): method test_full_loop_with_no_set_alpha_to_one (line 1463) | def test_full_loop_with_no_set_alpha_to_one(self): class ScoreSdeVeSchedulerTest (line 1473) | class ScoreSdeVeSchedulerTest(unittest.TestCase): method dummy_sample (line 1479) | def dummy_sample(self): method dummy_sample_deter (line 1490) | def dummy_sample_deter(self): method dummy_model (line 1504) | def dummy_model(self): method get_scheduler_config (line 1510) | def get_scheduler_config(self, **kwargs): method check_over_configs (line 1522) | def check_over_configs(self, time_step=0, **config): method check_over_forward (line 1552) | def check_over_forward(self, time_step=0, **forward_kwargs): method test_timesteps (line 1583) | def test_timesteps(self): method test_sigmas (line 1587) | def test_sigmas(self): method test_time_indices (line 1591) | def test_time_indices(self): method test_full_loop_no_noise (line 1595) | def test_full_loop_no_noise(self): method test_step_shape (line 1631) | def test_step_shape(self): class LMSDiscreteSchedulerTest (line 1655) | class LMSDiscreteSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 1659) | def get_scheduler_config(self, **kwargs): method test_timesteps (line 1670) | def test_timesteps(self): method test_betas (line 1674) | def test_betas(self): method test_schedules (line 1678) | def test_schedules(self): method test_prediction_type (line 1682) | def test_prediction_type(self): method test_time_indices (line 1686) | def test_time_indices(self): method test_full_loop_no_noise (line 1690) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 1714) | def test_full_loop_with_v_prediction(self): method test_full_loop_device (line 1738) | def test_full_loop_device(self): class EulerDiscreteSchedulerTest (line 1764) | class EulerDiscreteSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 1768) | def get_scheduler_config(self, **kwargs): method test_timesteps (line 1779) | def test_timesteps(self): method test_betas (line 1783) | def test_betas(self): method test_schedules (line 1787) | def test_schedules(self): method test_prediction_type (line 1791) | def test_prediction_type(self): method test_full_loop_no_noise (line 1795) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 1822) | def test_full_loop_with_v_prediction(self): method test_full_loop_device (line 1849) | def test_full_loop_device(self): class EulerAncestralDiscreteSchedulerTest (line 1877) | class EulerAncestralDiscreteSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 1881) | def get_scheduler_config(self, **kwargs): method test_timesteps (line 1892) | def test_timesteps(self): method test_betas (line 1896) | def test_betas(self): method test_schedules (line 1900) | def test_schedules(self): method test_prediction_type (line 1904) | def test_prediction_type(self): method test_full_loop_no_noise (line 1908) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 1935) | def test_full_loop_with_v_prediction(self): method test_full_loop_device (line 1962) | def test_full_loop_device(self): class IPNDMSchedulerTest (line 1989) | class IPNDMSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 1993) | def get_scheduler_config(self, **kwargs): method check_over_configs (line 1998) | def check_over_configs(self, time_step=0, **config): method test_from_save_pretrained (line 2032) | def test_from_save_pretrained(self): method check_over_forward (line 2035) | def check_over_forward(self, time_step=0, **forward_kwargs): method full_loop (line 2072) | def full_loop(self, **config): method test_step_shape (line 2092) | def test_step_shape(self): method test_timesteps (line 2128) | def test_timesteps(self): method test_inference_steps (line 2132) | def test_inference_steps(self): method test_full_loop_no_noise (line 2136) | def test_full_loop_no_noise(self): class VQDiffusionSchedulerTest (line 2143) | class VQDiffusionSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 2146) | def get_scheduler_config(self, **kwargs): method dummy_sample (line 2155) | def dummy_sample(self, num_vec_classes): method dummy_sample_deter (line 2165) | def dummy_sample_deter(self): method dummy_model (line 2168) | def dummy_model(self, num_vec_classes): method test_timesteps (line 2177) | def test_timesteps(self): method test_num_vec_classes (line 2181) | def test_num_vec_classes(self): method test_time_indices (line 2185) | def test_time_indices(self): method test_add_noise_device (line 2189) | def test_add_noise_device(self): class HeunDiscreteSchedulerTest (line 2193) | class HeunDiscreteSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 2197) | def get_scheduler_config(self, **kwargs): method test_timesteps (line 2208) | def test_timesteps(self): method test_betas (line 2212) | def test_betas(self): method test_schedules (line 2216) | def test_schedules(self): method test_prediction_type (line 2220) | def test_prediction_type(self): method test_full_loop_no_noise (line 2224) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 2254) | def test_full_loop_with_v_prediction(self): method test_full_loop_device (line 2284) | def test_full_loop_device(self): class KDPM2DiscreteSchedulerTest (line 2318) | class KDPM2DiscreteSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 2322) | def get_scheduler_config(self, **kwargs): method test_timesteps (line 2333) | def test_timesteps(self): method test_betas (line 2337) | def test_betas(self): method test_schedules (line 2341) | def test_schedules(self): method test_prediction_type (line 2345) | def test_prediction_type(self): method test_full_loop_with_v_prediction (line 2349) | def test_full_loop_with_v_prediction(self): method test_full_loop_no_noise (line 2379) | def test_full_loop_no_noise(self): method test_full_loop_device (line 2411) | def test_full_loop_device(self): class DEISMultistepSchedulerTest (line 2444) | class DEISMultistepSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 2448) | def get_scheduler_config(self, **kwargs): method check_over_configs (line 2460) | def check_over_configs(self, time_step=0, **config): method test_from_save_pretrained (line 2488) | def test_from_save_pretrained(self): method check_over_forward (line 2491) | def check_over_forward(self, time_step=0, **forward_kwargs): method full_loop (line 2520) | def full_loop(self, **config): method test_step_shape (line 2536) | def test_step_shape(self): method test_timesteps (line 2566) | def test_timesteps(self): method test_thresholding (line 2570) | def test_thresholding(self): method test_prediction_type (line 2585) | def test_prediction_type(self): method test_solver_order_and_type (line 2589) | def test_solver_order_and_type(self): method test_lower_order_final (line 2608) | def test_lower_order_final(self): method test_inference_steps (line 2612) | def test_inference_steps(self): method test_full_loop_no_noise (line 2616) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 2622) | def test_full_loop_with_v_prediction(self): method test_fp16_support (line 2628) | def test_fp16_support(self): class UniPCMultistepSchedulerTest (line 2645) | class UniPCMultistepSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 2649) | def get_scheduler_config(self, **kwargs): method check_over_configs (line 2662) | def check_over_configs(self, time_step=0, **config): method check_over_forward (line 2690) | def check_over_forward(self, time_step=0, **forward_kwargs): method full_loop (line 2719) | def full_loop(self, **config): method test_step_shape (line 2735) | def test_step_shape(self): method test_timesteps (line 2765) | def test_timesteps(self): method test_thresholding (line 2769) | def test_thresholding(self): method test_prediction_type (line 2783) | def test_prediction_type(self): method test_solver_order_and_type (line 2787) | def test_solver_order_and_type(self): method test_lower_order_final (line 2803) | def test_lower_order_final(self): method test_inference_steps (line 2807) | def test_inference_steps(self): method test_full_loop_no_noise (line 2811) | def test_full_loop_no_noise(self): method test_full_loop_with_v_prediction (line 2817) | def test_full_loop_with_v_prediction(self): method test_fp16_support (line 2823) | def test_fp16_support(self): class KDPM2AncestralDiscreteSchedulerTest (line 2840) | class KDPM2AncestralDiscreteSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 2844) | def get_scheduler_config(self, **kwargs): method test_timesteps (line 2855) | def test_timesteps(self): method test_betas (line 2859) | def test_betas(self): method test_schedules (line 2863) | def test_schedules(self): method test_full_loop_no_noise (line 2867) | def test_full_loop_no_noise(self): method test_prediction_type (line 2896) | def test_prediction_type(self): method test_full_loop_with_v_prediction (line 2900) | def test_full_loop_with_v_prediction(self): method test_full_loop_device (line 2929) | def test_full_loop_device(self): class UnCLIPSchedulerTest (line 2958) | class UnCLIPSchedulerTest(SchedulerCommonTest): method get_scheduler_config (line 2961) | def get_scheduler_config(self, **kwargs): method test_timesteps (line 2973) | def test_timesteps(self): method test_variance_type (line 2977) | def test_variance_type(self): method test_clip_sample (line 2981) | def test_clip_sample(self): method test_clip_sample_range (line 2985) | def test_clip_sample_range(self): method test_prediction_type (line 2989) | def test_prediction_type(self): method test_time_indices (line 2993) | def test_time_indices(self): method test_variance_fixed_small_log (line 3001) | def test_variance_fixed_small_log(self): method test_variance_learned_range (line 3010) | def test_variance_learned_range(self): method test_full_loop (line 3021) | def test_full_loop(self): method test_full_loop_skip_timesteps (line 3047) | def test_full_loop_skip_timesteps(self): method test_trained_betas (line 3082) | def test_trained_betas(self): method test_add_noise_device (line 3085) | def test_add_noise_device(self): FILE: tests/test_scheduler_flax.py class FlaxSchedulerCommonTest (line 34) | class FlaxSchedulerCommonTest(unittest.TestCase): method dummy_sample (line 39) | def dummy_sample(self): method dummy_sample_deter (line 51) | def dummy_sample_deter(self): method get_scheduler_config (line 63) | def get_scheduler_config(self): method dummy_model (line 66) | def dummy_model(self): method check_over_configs (line 72) | def check_over_configs(self, time_step=0, **config): method check_over_forward (line 100) | def check_over_forward(self, time_step=0, **forward_kwargs): method test_from_save_pretrained (line 129) | def test_from_save_pretrained(self): method test_step_shape (line 157) | def test_step_shape(self): method test_scheduler_outputs_equivalence (line 181) | def test_scheduler_outputs_equivalence(self): method test_deprecated_kwargs (line 232) | def test_deprecated_kwargs(self): class FlaxDDPMSchedulerTest (line 255) | class FlaxDDPMSchedulerTest(FlaxSchedulerCommonTest): method get_scheduler_config (line 258) | def get_scheduler_config(self, **kwargs): method test_timesteps (line 271) | def test_timesteps(self): method test_betas (line 275) | def test_betas(self): method test_schedules (line 279) | def test_schedules(self): method test_variance_type (line 283) | def test_variance_type(self): method test_clip_sample (line 287) | def test_clip_sample(self): method test_time_indices (line 291) | def test_time_indices(self): method test_variance (line 295) | def test_variance(self): method test_full_loop_no_noise (line 305) | def test_full_loop_no_noise(self): class FlaxDDIMSchedulerTest (line 346) | class FlaxDDIMSchedulerTest(FlaxSchedulerCommonTest): method get_scheduler_config (line 350) | def get_scheduler_config(self, **kwargs): method full_loop (line 361) | def full_loop(self, **config): method check_over_configs (line 384) | def check_over_configs(self, time_step=0, **config): method test_from_save_pretrained (line 412) | def test_from_save_pretrained(self): method check_over_forward (line 440) | def check_over_forward(self, time_step=0, **forward_kwargs): method test_scheduler_outputs_equivalence (line 469) | def test_scheduler_outputs_equivalence(self): method test_step_shape (line 520) | def test_step_shape(self): method test_timesteps (line 544) | def test_timesteps(self): method test_steps_offset (line 548) | def test_steps_offset(self): method test_betas (line 559) | def test_betas(self): method test_schedules (line 563) | def test_schedules(self): method test_time_indices (line 567) | def test_time_indices(self): method test_inference_steps (line 571) | def test_inference_steps(self): method test_variance (line 575) | def test_variance(self): method test_full_loop_no_noise (line 588) | def test_full_loop_no_noise(self): method test_full_loop_with_set_alpha_to_one (line 597) | def test_full_loop_with_set_alpha_to_one(self): method test_full_loop_with_no_set_alpha_to_one (line 610) | def test_full_loop_with_no_set_alpha_to_one(self): method test_prediction_type (line 625) | def test_prediction_type(self): class FlaxPNDMSchedulerTest (line 631) | class FlaxPNDMSchedulerTest(FlaxSchedulerCommonTest): method get_scheduler_config (line 635) | def get_scheduler_config(self, **kwargs): method check_over_configs (line 646) | def check_over_configs(self, time_step=0, **config): method test_from_save_pretrained (line 678) | def test_from_save_pretrained(self): method test_scheduler_outputs_equivalence (line 681) | def test_scheduler_outputs_equivalence(self): method check_over_forward (line 732) | def check_over_forward(self, time_step=0, **forward_kwargs): method full_loop (line 767) | def full_loop(self, **config): method test_step_shape (line 788) | def test_step_shape(self): method test_timesteps (line 822) | def test_timesteps(self): method test_steps_offset (line 826) | def test_steps_offset(self): method test_betas (line 840) | def test_betas(self): method test_schedules (line 844) | def test_schedules(self): method test_time_indices (line 848) | def test_time_indices(self): method test_inference_steps (line 852) | def test_inference_steps(self): method test_pow_of_3_inference_steps (line 856) | def test_pow_of_3_inference_steps(self): method test_inference_plms_no_past_residuals (line 874) | def test_inference_plms_no_past_residuals(self): method test_full_loop_no_noise (line 883) | def test_full_loop_no_noise(self): method test_full_loop_with_set_alpha_to_one (line 895) | def test_full_loop_with_set_alpha_to_one(self): method test_full_loop_with_no_set_alpha_to_one (line 908) | def test_full_loop_with_no_set_alpha_to_one(self): FILE: tests/test_training.py class TrainingTests (line 28) | class TrainingTests(unittest.TestCase): method get_model_optimizer (line 29) | def get_model_optimizer(self, resolution=32): method test_training_step_equality (line 36) | def test_training_step_equality(self): FILE: tests/test_unet_2d_blocks.py class DownBlock2DTests (line 23) | class DownBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method test_output (line 27) | def test_output(self): class ResnetDownsampleBlock2DTests (line 32) | class ResnetDownsampleBlock2DTests(UNetBlockTesterMixin, unittest.TestCa... method test_output (line 36) | def test_output(self): class AttnDownBlock2DTests (line 41) | class AttnDownBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method test_output (line 45) | def test_output(self): class CrossAttnDownBlock2DTests (line 50) | class CrossAttnDownBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method prepare_init_args_and_inputs_for_common (line 54) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 59) | def test_output(self): class SimpleCrossAttnDownBlock2DTests (line 64) | class SimpleCrossAttnDownBlock2DTests(UNetBlockTesterMixin, unittest.Tes... method dummy_input (line 69) | def dummy_input(self): method prepare_init_args_and_inputs_for_common (line 72) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 78) | def test_output(self): class SkipDownBlock2DTests (line 83) | class SkipDownBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 88) | def dummy_input(self): method test_output (line 91) | def test_output(self): class AttnSkipDownBlock2DTests (line 96) | class AttnSkipDownBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 101) | def dummy_input(self): method test_output (line 104) | def test_output(self): class DownEncoderBlock2DTests (line 109) | class DownEncoderBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 114) | def dummy_input(self): method prepare_init_args_and_inputs_for_common (line 117) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 125) | def test_output(self): class AttnDownEncoderBlock2DTests (line 130) | class AttnDownEncoderBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 135) | def dummy_input(self): method prepare_init_args_and_inputs_for_common (line 138) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 146) | def test_output(self): class UNetMidBlock2DTests (line 151) | class UNetMidBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method prepare_init_args_and_inputs_for_common (line 155) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 163) | def test_output(self): class UNetMidBlock2DCrossAttnTests (line 168) | class UNetMidBlock2DCrossAttnTests(UNetBlockTesterMixin, unittest.TestCa... method prepare_init_args_and_inputs_for_common (line 172) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 177) | def test_output(self): class UNetMidBlock2DSimpleCrossAttnTests (line 182) | class UNetMidBlock2DSimpleCrossAttnTests(UNetBlockTesterMixin, unittest.... method dummy_input (line 187) | def dummy_input(self): method prepare_init_args_and_inputs_for_common (line 190) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 195) | def test_output(self): class UpBlock2DTests (line 200) | class UpBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 205) | def dummy_input(self): method test_output (line 208) | def test_output(self): class ResnetUpsampleBlock2DTests (line 213) | class ResnetUpsampleBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 218) | def dummy_input(self): method test_output (line 221) | def test_output(self): class CrossAttnUpBlock2DTests (line 226) | class CrossAttnUpBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 231) | def dummy_input(self): method prepare_init_args_and_inputs_for_common (line 234) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 239) | def test_output(self): class SimpleCrossAttnUpBlock2DTests (line 244) | class SimpleCrossAttnUpBlock2DTests(UNetBlockTesterMixin, unittest.TestC... method dummy_input (line 249) | def dummy_input(self): method prepare_init_args_and_inputs_for_common (line 252) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 257) | def test_output(self): class AttnUpBlock2DTests (line 265) | class AttnUpBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 270) | def dummy_input(self): method test_output (line 274) | def test_output(self): class SkipUpBlock2DTests (line 279) | class SkipUpBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 284) | def dummy_input(self): method test_output (line 287) | def test_output(self): class AttnSkipUpBlock2DTests (line 292) | class AttnSkipUpBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 297) | def dummy_input(self): method test_output (line 300) | def test_output(self): class UpDecoderBlock2DTests (line 305) | class UpDecoderBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 310) | def dummy_input(self): method prepare_init_args_and_inputs_for_common (line 313) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 319) | def test_output(self): class AttnUpDecoderBlock2DTests (line 324) | class AttnUpDecoderBlock2DTests(UNetBlockTesterMixin, unittest.TestCase): method dummy_input (line 329) | def dummy_input(self): method prepare_init_args_and_inputs_for_common (line 332) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 338) | def test_output(self): FILE: tests/test_unet_blocks_common.py class UNetBlockTesterMixin (line 25) | class UNetBlockTesterMixin: method dummy_input (line 27) | def dummy_input(self): method output_shape (line 31) | def output_shape(self): method get_dummy_input (line 41) | def get_dummy_input( method prepare_init_args_and_inputs_for_common (line 74) | def prepare_init_args_and_inputs_for_common(self): method test_output (line 89) | def test_output(self, expected_slice): method test_training (line 108) | def test_training(self): FILE: tests/test_utils.py class DeprecateTester (line 22) | class DeprecateTester(unittest.TestCase): method test_deprecate_function_arg (line 26) | def test_deprecate_function_arg(self): method test_deprecate_function_arg_tuple (line 39) | def test_deprecate_function_arg_tuple(self): method test_deprecate_function_args (line 52) | def test_deprecate_function_args(self): method test_deprecate_function_incorrect_arg (line 73) | def test_deprecate_function_incorrect_arg(self): method test_deprecate_arg_no_kwarg (line 83) | def test_deprecate_arg_no_kwarg(self): method test_deprecate_args_no_kwarg (line 92) | def test_deprecate_args_no_kwarg(self): method test_deprecate_class_obj (line 107) | def test_deprecate_class_obj(self): method test_deprecate_class_objs (line 120) | def test_deprecate_class_objs(self): method test_deprecate_incorrect_version (line 148) | def test_deprecate_incorrect_version(self): method test_deprecate_incorrect_no_standard_warn (line 160) | def test_deprecate_incorrect_no_standard_warn(self): method test_deprecate_stacklevel (line 166) | def test_deprecate_stacklevel(self): FILE: utils/check_config_docstrings.py function check_config_docstrings_have_checkpoints (line 53) | def check_config_docstrings_have_checkpoints(): FILE: utils/check_copies.py function _should_continue (line 41) | def _should_continue(line, indent): function find_code_in_diffusers (line 45) | def find_code_in_diffusers(object_name): function get_indent (line 93) | def get_indent(code): function blackify (line 103) | def blackify(code): function is_copy_consistent (line 116) | def is_copy_consistent(filename, overwrite=False): function check_copies (line 193) | def check_copies(overwrite: bool = False): FILE: utils/check_doc_toc.py function clean_doc_toc (line 25) | def clean_doc_toc(doc_list): function check_scheduler_doc (line 70) | def check_scheduler_doc(overwrite=False): function check_pipeline_doc (line 105) | def check_pipeline_doc(overwrite=False): FILE: utils/check_dummies.py function find_backend (line 58) | def find_backend(line): function read_init (line 67) | def read_init(): function create_dummy_object (line 102) | def create_dummy_object(name, backend_name): function create_dummy_files (line 112) | def create_dummy_files(backend_specific_objects=None): function check_dummies (line 129) | def check_dummies(overwrite=False): FILE: utils/check_inits.py function find_backend (line 50) | def find_backend(line): function parse_init (line 59) | def parse_init(init_file): function analyze_results (line 190) | def analyze_results(import_dict_objects, type_hint_objects): function check_all_inits (line 222) | def check_all_inits(): function get_transformers_submodules (line 241) | def get_transformers_submodules(): function check_submodules (line 274) | def check_submodules(): FILE: utils/check_repo.py function check_model_list (line 209) | def check_model_list(): function get_model_modules (line 231) | def get_model_modules(): function get_models (line 270) | def get_models(module, include_pretrained=False): function is_a_private_model (line 283) | def is_a_private_model(model): function check_models_are_in_init (line 298) | def check_models_are_in_init(): function get_model_test_files (line 315) | def get_model_test_files(): function find_tested_models (line 354) | def find_tested_models(test_file): function check_models_are_tested (line 372) | def check_models_are_tested(module, test_file): function check_all_models_are_tested (line 397) | def check_all_models_are_tested(): function get_all_auto_configured_models (line 417) | def get_all_auto_configured_models(): function ignore_unautoclassed (line 435) | def ignore_unautoclassed(model_name): function check_models_are_auto_configured (line 446) | def check_models_are_auto_configured(module, all_auto_models): function check_all_models_are_auto_configured (line 460) | def check_all_models_are_auto_configured(): function check_decorator_order (line 497) | def check_decorator_order(filename): function check_all_decorator_order (line 515) | def check_all_decorator_order(): function find_all_documented_objects (line 531) | def find_all_documented_objects(): function ignore_undocumented (line 621) | def ignore_undocumented(name): function check_all_objects_are_documented (line 659) | def check_all_objects_are_documented(): function check_model_type_doc_match (line 674) | def check_model_type_doc_match(): function is_rst_docstring (line 709) | def is_rst_docstring(docstring): function check_docstrings_are_in_md (line 722) | def check_docstrings_are_in_md(): function check_repo_quality (line 745) | def check_repo_quality(): FILE: utils/check_table.py function _find_text_in_file (line 30) | def _find_text_in_file(filename, start_prompt, end_prompt): function camel_case_split (line 75) | def camel_case_split(identifier): function _center_text (line 81) | def _center_text(text, width): function get_model_table_from_auto_modules (line 88) | def get_model_table_from_auto_modules(): function check_model_table (line 161) | def check_model_table(overwrite=False): FILE: utils/custom_init_isort.py function get_indent (line 35) | def get_indent(line): function split_code_in_indented_blocks (line 41) | def split_code_in_indented_blocks(code, indent_level="", start_prompt=No... function ignore_underscore (line 88) | def ignore_underscore(key): function sort_objects (line 97) | def sort_objects(objects, key=None): function sort_objects_in_import (line 117) | def sort_objects_in_import(import_statement): function sort_imports (line 168) | def sort_imports(file, check_only=True): function sort_imports_in_all_inits (line 236) | def sort_imports_in_all_inits(check_only=True): FILE: utils/overwrite_expected_slice.py function overwrite_file (line 19) | def overwrite_file(file, class_name, test_name, correct_line, done_test): function main (line 67) | def main(correct, fail=None): FILE: utils/release.py function update_version_in_file (line 37) | def update_version_in_file(fname, version, pattern): function update_version_in_examples (line 48) | def update_version_in_examples(version): function global_version_update (line 61) | def global_version_update(version, patch=False): function clean_main_ref_in_model_list (line 69) | def clean_main_ref_in_model_list(): function get_version (line 97) | def get_version(): function pre_release_work (line 105) | def pre_release_work(patch=False): function post_release_work (line 132) | def post_release_work(): FILE: utils/stale.py function main (line 35) | def main():