SYMBOL INDEX (581 symbols across 157 files) FILE: src/invoke_training/_shared/accelerator/accelerator_utils.py function initialize_accelerator (line 14) | def initialize_accelerator( function initialize_logging (line 40) | def initialize_logging(logger_name: str, accelerator: Accelerator) -> Mu... function get_mixed_precision_dtype (line 71) | def get_mixed_precision_dtype(accelerator: Accelerator): function get_dtype_from_str (line 95) | def get_dtype_from_str(dtype_str: Literal["float16", "bfloat16", "float3... FILE: src/invoke_training/_shared/checkpoints/checkpoint_tracker.py class CheckpointTracker (line 6) | class CheckpointTracker: method __init__ (line 14) | def __init__( method prune (line 48) | def prune(self, buffer_num: int = 1) -> int: method get_path (line 83) | def get_path(self, epoch: int, step: int) -> str: FILE: src/invoke_training/_shared/checkpoints/lora_checkpoint_utils.py function save_multi_model_peft_checkpoint (line 7) | def save_multi_model_peft_checkpoint(checkpoint_dir: Path | str, models:... function load_multi_model_peft_checkpoint (line 30) | def load_multi_model_peft_checkpoint( function _convert_peft_state_dict_to_kohya_state_dict (line 57) | def _convert_peft_state_dict_to_kohya_state_dict( function _convert_peft_models_to_kohya_state_dict (line 79) | def _convert_peft_models_to_kohya_state_dict( FILE: src/invoke_training/_shared/checkpoints/serialization.py function save_state_dict (line 8) | def save_state_dict(state_dict: typing.Dict[str, torch.Tensor], out_file... function load_state_dict (line 33) | def load_state_dict(in_file: typing.Union[Path, str]) -> typing.Dict[str... FILE: src/invoke_training/_shared/data/data_loaders/dreambooth_sd_dataloader.py function build_dreambooth_sd_dataloader (line 25) | def build_dreambooth_sd_dataloader( FILE: src/invoke_training/_shared/data/data_loaders/image_caption_flux_dataloader.py function build_image_caption_flux_dataloader (line 33) | def build_image_caption_flux_dataloader( # noqa: C901 FILE: src/invoke_training/_shared/data/data_loaders/image_caption_sd_dataloader.py function sd_image_caption_collate_fn (line 27) | def sd_image_caption_collate_fn(examples): function build_aspect_ratio_bucket_manager (line 64) | def build_aspect_ratio_bucket_manager(config: AspectRatioBucketConfig): function build_image_caption_sd_dataloader (line 73) | def build_image_caption_sd_dataloader( # noqa: C901 FILE: src/invoke_training/_shared/data/data_loaders/image_pair_preference_sd_dataloader.py function sd_image_pair_preference_collate_fn (line 15) | def sd_image_pair_preference_collate_fn(examples): function build_image_pair_preference_sd_dataloader (line 49) | def build_image_pair_preference_sd_dataloader( FILE: src/invoke_training/_shared/data/data_loaders/textual_inversion_sd_dataloader.py function get_preset_ti_caption_templates (line 33) | def get_preset_ti_caption_templates(preset: Literal["object", "style"]) ... function build_textual_inversion_sd_dataloader (line 97) | def build_textual_inversion_sd_dataloader( # noqa: C901 FILE: src/invoke_training/_shared/data/datasets/build_dataset.py function build_hf_hub_image_caption_dataset (line 15) | def build_hf_hub_image_caption_dataset(config: HFHubImageCaptionDatasetC... function build_image_caption_jsonl_dataset (line 27) | def build_image_caption_jsonl_dataset(config: ImageCaptionJsonlDatasetCo... function build_image_caption_dir_dataset (line 36) | def build_image_caption_dir_dataset(config: ImageCaptionDirDatasetConfig... function build_hf_image_pair_preference_dataset (line 43) | def build_hf_image_pair_preference_dataset( FILE: src/invoke_training/_shared/data/datasets/hf_image_caption_dataset.py class HFImageCaptionDataset (line 11) | class HFImageCaptionDataset(torch.utils.data.Dataset): method __init__ (line 18) | def __init__(self, hf_dataset, image_column: str = "image", caption_co... method from_dir (line 42) | def from_dir( method from_hub (line 67) | def from_hub( method get_image_dimensions (line 87) | def get_image_dimensions(self) -> list[Resolution]: method __len__ (line 101) | def __len__(self) -> int: method __getitem__ (line 109) | def __getitem__(self, idx: int) -> typing.Dict[str, typing.Any]: FILE: src/invoke_training/_shared/data/datasets/hf_image_pair_preference_dataset.py class HFImagePairPreferenceDataset (line 9) | class HFImagePairPreferenceDataset(torch.utils.data.Dataset): method __init__ (line 16) | def __init__( method from_hub (line 77) | def from_hub( method __len__ (line 101) | def __len__(self) -> int: method __getitem__ (line 109) | def __getitem__(self, idx: int) -> typing.Dict[str, typing.Any]: FILE: src/invoke_training/_shared/data/datasets/image_caption_dir_dataset.py class ImageCaptionDirDataset (line 10) | class ImageCaptionDirDataset(torch.utils.data.Dataset): method __init__ (line 13) | def __init__( method _load_image (line 64) | def _load_image(self, image_path: str) -> Image.Image: method get_image_dimensions (line 69) | def get_image_dimensions(self) -> list[Resolution]: method __len__ (line 83) | def __len__(self) -> int: method __getitem__ (line 86) | def __getitem__(self, idx: int) -> typing.Dict[str, typing.Any]: FILE: src/invoke_training/_shared/data/datasets/image_caption_jsonl_dataset.py class ImageCaptionExample (line 16) | class ImageCaptionExample(BaseModel): class ImageCaptionJsonlDataset (line 22) | class ImageCaptionJsonlDataset(torch.utils.data.Dataset): method __init__ (line 25) | def __init__( method save_jsonl (line 55) | def save_jsonl(self): method _get_image_path (line 67) | def _get_image_path(self, idx: int) -> str: method _get_mask_path (line 77) | def _get_mask_path(self, idx: int) -> str: method _load_image (line 87) | def _load_image(self, image_path: str) -> Image.Image: method _load_mask (line 92) | def _load_mask(self, mask_path: str) -> Image.Image: method _load_example (line 95) | def _load_example(self, idx: int) -> dict[str, typing.Any]: method get_image_dimensions (line 105) | def get_image_dimensions(self) -> list[Resolution]: method __len__ (line 118) | def __len__(self) -> int: method __getitem__ (line 121) | def __getitem__(self, idx: int) -> typing.Dict[str, typing.Any]: FILE: src/invoke_training/_shared/data/datasets/image_dir_dataset.py class ImageDirDataset (line 10) | class ImageDirDataset(torch.utils.data.Dataset): method __init__ (line 13) | def __init__( method _load_image (line 49) | def _load_image(self, image_path: str) -> Image.Image: method get_image_dimensions (line 54) | def get_image_dimensions(self) -> list[Resolution]: method __len__ (line 68) | def __len__(self) -> int: method __getitem__ (line 71) | def __getitem__(self, idx: int) -> typing.Dict[str, typing.Any]: FILE: src/invoke_training/_shared/data/datasets/image_pair_preference_dataset.py class ImagePairPreferenceDataset (line 11) | class ImagePairPreferenceDataset(torch.utils.data.Dataset): method __init__ (line 12) | def __init__(self, dataset_dir: str): method save_metadata (line 19) | def save_metadata( method __len__ (line 27) | def __len__(self) -> int: method __getitem__ (line 30) | def __getitem__(self, idx: int) -> typing.Dict[str, typing.Any]: FILE: src/invoke_training/_shared/data/datasets/transform_dataset.py class TransformDataset (line 11) | class TransformDataset(torch.utils.data.Dataset): method __init__ (line 14) | def __init__(self, base_dataset: torch.utils.data.Dataset, transforms:... method __len__ (line 19) | def __len__(self) -> int: method __getitem__ (line 22) | def __getitem__(self, idx: int) -> DataType: FILE: src/invoke_training/_shared/data/samplers/aspect_ratio_bucket_batch_sampler.py class AspectRatioBucketBatchSampler (line 15) | class AspectRatioBucketBatchSampler(Sampler[list[int]]): method __init__ (line 18) | def __init__( method __str__ (line 34) | def __str__(self) -> str: method from_image_sizes (line 44) | def from_image_sizes( method _build_bucket_to_index_map (line 57) | def _build_bucket_to_index_map( method get_buckets (line 73) | def get_buckets(self) -> AspectRatioBuckets: method __iter__ (line 76) | def __iter__(self) -> Iterator[list[int]]: method __len__ (line 103) | def __len__(self) -> int: function log_aspect_ratio_buckets (line 110) | def log_aspect_ratio_buckets(logger: logging.Logger, batch_sampler: Aspe... FILE: src/invoke_training/_shared/data/samplers/batch_offset_sampler.py class BatchOffsetSampler (line 6) | class BatchOffsetSampler(Sampler[int]): method __init__ (line 9) | def __init__(self, sampler: Sampler[int], offset: int): method __iter__ (line 13) | def __iter__(self) -> typing.Iterator[int]: method __len__ (line 18) | def __len__(self) -> int: FILE: src/invoke_training/_shared/data/samplers/concat_sampler.py class ConcatSampler (line 9) | class ConcatSampler(Sampler[T_co]): method __init__ (line 19) | def __init__(self, samplers: list[Sampler[T_co] | typing.Iterable[T_co... method __iter__ (line 22) | def __iter__(self) -> typing.Iterator[T_co]: method __len__ (line 25) | def __len__(self) -> int: FILE: src/invoke_training/_shared/data/samplers/interleaved_sampler.py class InterleavedSampler (line 8) | class InterleavedSampler(Sampler[T_co]): method __init__ (line 21) | def __init__(self, samplers: list[Sampler[T_co] | typing.Iterable[T_co... method __iter__ (line 25) | def __iter__(self) -> typing.Iterator[T_co]: method __len__ (line 38) | def __len__(self) -> int: FILE: src/invoke_training/_shared/data/samplers/offset_sampler.py class OffsetSampler (line 6) | class OffsetSampler(Sampler[int]): method __init__ (line 9) | def __init__(self, sampler: Sampler[int], offset: int): method __iter__ (line 13) | def __iter__(self) -> typing.Iterator[int]: method __len__ (line 17) | def __len__(self) -> int: FILE: src/invoke_training/_shared/data/transforms/caption_prefix_transform.py class CaptionPrefixTransform (line 4) | class CaptionPrefixTransform: method __init__ (line 7) | def __init__(self, caption_field_name: str, prefix: str): method __call__ (line 11) | def __call__(self, data: typing.Dict[str, typing.Any]) -> typing.Dict[... FILE: src/invoke_training/_shared/data/transforms/concat_fields_transform.py class ConcatFieldsTransform (line 4) | class ConcatFieldsTransform: method __init__ (line 7) | def __init__(self, src_field_names: list[str], dst_field_name: str, se... method __call__ (line 12) | def __call__(self, data: typing.Dict[str, typing.Any]) -> typing.Dict[... FILE: src/invoke_training/_shared/data/transforms/constant_field_transform.py class ConstantFieldTransform (line 4) | class ConstantFieldTransform: method __init__ (line 7) | def __init__(self, field_name: str, field_value: typing.Any): method __call__ (line 11) | def __call__(self, data: typing.Dict[str, typing.Any]) -> typing.Dict[... FILE: src/invoke_training/_shared/data/transforms/drop_field_transform.py class DropFieldTransform (line 4) | class DropFieldTransform: method __init__ (line 7) | def __init__(self, field_to_drop: str): method __call__ (line 10) | def __call__(self, data: typing.Dict[str, typing.Any]) -> typing.Dict[... FILE: src/invoke_training/_shared/data/transforms/flux_image_transform.py class FluxImageTransform (line 10) | class FluxImageTransform: method __init__ (line 13) | def __init__( method __call__ (line 42) | def __call__(self, data: typing.Dict[str, typing.Any]) -> typing.Dict[... FILE: src/invoke_training/_shared/data/transforms/load_cache_transform.py class LoadCacheTransform (line 6) | class LoadCacheTransform: method __init__ (line 9) | def __init__( method __call__ (line 24) | def __call__(self, data: typing.Dict[str, typing.Any]) -> typing.Dict[... FILE: src/invoke_training/_shared/data/transforms/sd_image_transform.py class SDImageTransform (line 11) | class SDImageTransform: method __init__ (line 14) | def __init__( method __call__ (line 59) | def __call__(self, data: typing.Dict[str, typing.Any]) -> typing.Dict[... FILE: src/invoke_training/_shared/data/transforms/shuffle_caption_transform.py class ShuffleCaptionTransform (line 6) | class ShuffleCaptionTransform: method __init__ (line 14) | def __init__(self, field_name: str, delimiter: str = ",", seed: int = 0): method __call__ (line 19) | def __call__(self, data: typing.Dict[str, typing.Any]) -> typing.Dict[... FILE: src/invoke_training/_shared/data/transforms/template_caption_transform.py class TemplateCaptionTransform (line 6) | class TemplateCaptionTransform: method __init__ (line 11) | def __init__(self, field_name: str, placeholder_str: str, caption_temp... method __call__ (line 17) | def __call__(self, data: typing.Dict[str, typing.Any]) -> typing.Dict[... FILE: src/invoke_training/_shared/data/transforms/tensor_disk_cache.py class TensorDiskCache (line 7) | class TensorDiskCache: method __init__ (line 10) | def __init__(self, cache_dir: str): method _get_path (line 16) | def _get_path(self, key: int): method save (line 25) | def save(self, key: int, data: typing.Dict[str, torch.Tensor]): method load (line 41) | def load(self, key: int) -> typing.Dict[str, torch.Tensor]: FILE: src/invoke_training/_shared/data/utils/aspect_ratio_bucket_manager.py class AspectRatioBucketManager (line 4) | class AspectRatioBucketManager: method __init__ (line 5) | def __init__(self, buckets: set[Resolution]): method from_constraints (line 9) | def from_constraints(cls, target_resolution: int, start_dim: int, end_... method build_aspect_ratio_buckets (line 19) | def build_aspect_ratio_buckets( method get_aspect_ratio_bucket (line 56) | def get_aspect_ratio_bucket(self, resolution: Resolution): FILE: src/invoke_training/_shared/data/utils/resize.py function resize_to_cover (line 9) | def resize_to_cover(image: Image, size_to_cover: Resolution) -> Image: FILE: src/invoke_training/_shared/data/utils/resolution.py class Resolution (line 4) | class Resolution: method __init__ (line 5) | def __init__(self, height: int, width: int): method parse (line 10) | def parse(cls, resolution: Union[int, tuple[int, int], "Resolution"]): method aspect_ratio (line 23) | def aspect_ratio(self): method to_tuple (line 26) | def to_tuple(self) -> tuple[int, int]: method __eq__ (line 29) | def __eq__(self, other: "Resolution") -> bool: method __lt__ (line 32) | def __lt__(self, other: "Resolution") -> bool: method __hash__ (line 35) | def __hash__(self): FILE: src/invoke_training/_shared/flux/encoding_utils.py function get_clip_prompt_embeds (line 8) | def get_clip_prompt_embeds( function get_t5_prompt_embeds (line 55) | def get_t5_prompt_embeds( function handle_lora_scale (line 101) | def handle_lora_scale( function reset_lora_scale (line 121) | def reset_lora_scale( function encode_prompt (line 141) | def encode_prompt( FILE: src/invoke_training/_shared/flux/lora_checkpoint_utils.py function save_flux_peft_checkpoint (line 62) | def save_flux_peft_checkpoint( function load_flux_peft_checkpoint (line 79) | def load_flux_peft_checkpoint( function save_flux_kohya_checkpoint (line 100) | def save_flux_kohya_checkpoint( function convert_flux_peft_checkpoint_to_kohya_state_dict (line 121) | def convert_flux_peft_checkpoint_to_kohya_state_dict( function _convert_peft_models_to_kohya_state_dict (line 158) | def _convert_peft_models_to_kohya_state_dict( function find_matching_key_prefix (line 185) | def find_matching_key_prefix(state_dict, key_pattern): function convert_layer_weights (line 204) | def convert_layer_weights(target_dict, source_dict, source_pattern, targ... function convert_double_transformer_block (line 231) | def convert_double_transformer_block(target_dict, source_dict, prefix=""... function convert_single_transformer_block (line 333) | def convert_single_transformer_block(target_dict, source_dict, prefix, b... function convert_embedding_layers (line 384) | def convert_embedding_layers(target_dict, source_dict, prefix, has_guida... function convert_output_layers (line 436) | def convert_output_layers(target_dict, source_dict, prefix): function convert_diffusers_to_flux_transformer_checkpoint (line 460) | def convert_diffusers_to_flux_transformer_checkpoint( FILE: src/invoke_training/_shared/flux/model_loading_utils.py class PipelineVersionEnum (line 9) | class PipelineVersionEnum(Enum): function load_pipeline (line 13) | def load_pipeline( function load_models_flux (line 65) | def load_models_flux( FILE: src/invoke_training/_shared/flux/validation.py function generate_validation_images_flux (line 25) | def generate_validation_images_flux( # noqa: C901 FILE: src/invoke_training/_shared/optimizer/optimizer_utils.py function initialize_optimizer (line 7) | def initialize_optimizer( FILE: src/invoke_training/_shared/stable_diffusion/base_model_version.py class BaseModelVersionEnum (line 6) | class BaseModelVersionEnum(Enum): function get_base_model_version (line 13) | def get_base_model_version( function check_base_model_version (line 54) | def check_base_model_version( FILE: src/invoke_training/_shared/stable_diffusion/checkpoint_utils.py function save_sdxl_diffusers_unet_checkpoint (line 8) | def save_sdxl_diffusers_unet_checkpoint( function save_sdxl_diffusers_checkpoint (line 25) | def save_sdxl_diffusers_checkpoint( FILE: src/invoke_training/_shared/stable_diffusion/lora_checkpoint_utils.py function save_sd_peft_checkpoint (line 66) | def save_sd_peft_checkpoint( function load_sd_peft_checkpoint (line 78) | def load_sd_peft_checkpoint( function save_sdxl_peft_checkpoint (line 91) | def save_sdxl_peft_checkpoint( function load_sdxl_peft_checkpoint (line 108) | def load_sdxl_peft_checkpoint( function save_sd_kohya_checkpoint (line 129) | def save_sd_kohya_checkpoint(checkpoint_path: Path, unet: peft.PeftModel... function save_sdxl_kohya_checkpoint (line 143) | def save_sdxl_kohya_checkpoint( function convert_sd_peft_checkpoint_to_kohya_state_dict (line 165) | def convert_sd_peft_checkpoint_to_kohya_state_dict( FILE: src/invoke_training/_shared/stable_diffusion/min_snr_weighting.py function compute_snr (line 5) | def compute_snr(noise_scheduler: DDPMScheduler, timesteps: torch.Tensor): FILE: src/invoke_training/_shared/stable_diffusion/model_loading_utils.py class PipelineVersionEnum (line 21) | class PipelineVersionEnum(Enum): function load_pipeline (line 26) | def load_pipeline( function from_pretrained_with_variant_fallback (line 75) | def from_pretrained_with_variant_fallback( function load_models_sd (line 113) | def load_models_sd( function load_models_sdxl (line 167) | def load_models_sdxl( FILE: src/invoke_training/_shared/stable_diffusion/textual_inversion.py function _expand_placeholder_token (line 10) | def _expand_placeholder_token(placeholder_token: str, num_vectors: int =... function _add_tokens_to_tokenizer (line 23) | def _add_tokens_to_tokenizer(placeholder_tokens: list[str], tokenizer: P... function expand_placeholders_in_caption (line 37) | def expand_placeholders_in_caption(caption: str, tokenizer: CLIPTokenize... function initialize_placeholder_tokens_from_initializer_token (line 68) | def initialize_placeholder_tokens_from_initializer_token( function initialize_placeholder_tokens_from_initial_phrase (line 106) | def initialize_placeholder_tokens_from_initial_phrase( function initialize_placeholder_tokens_from_initial_embedding (line 131) | def initialize_placeholder_tokens_from_initial_embedding( function restore_original_embeddings (line 172) | def restore_original_embeddings( FILE: src/invoke_training/_shared/stable_diffusion/tokenize_captions.py function tokenize_captions (line 7) | def tokenize_captions(tokenizer: CLIPTokenizer, captions: list[str]) -> ... FILE: src/invoke_training/_shared/stable_diffusion/validation.py function generate_validation_images_sd (line 24) | def generate_validation_images_sd( # noqa: C901 function generate_validation_images_sdxl (line 140) | def generate_validation_images_sdxl( # noqa: C901 FILE: src/invoke_training/_shared/tools/generate_images.py function generate_images (line 15) | def generate_images( FILE: src/invoke_training/_shared/utils/import_xformers.py function import_xformers (line 1) | def import_xformers(): FILE: src/invoke_training/_shared/utils/jsonl.py function load_jsonl (line 6) | def load_jsonl(jsonl_path: Path | str) -> list[Any]: function save_jsonl (line 15) | def save_jsonl(data: list[Any], jsonl_path: Path | str) -> None: FILE: src/invoke_training/config/base_pipeline_config.py class BasePipelineConfig (line 7) | class BasePipelineConfig(ConfigBaseModel): FILE: src/invoke_training/config/config_base_model.py class ConfigBaseModel (line 4) | class ConfigBaseModel(BaseModel): FILE: src/invoke_training/config/data/data_loader_config.py class AspectRatioBucketConfig (line 10) | class AspectRatioBucketConfig(ConfigBaseModel): class ImageCaptionSDDataLoaderConfig (line 42) | class ImageCaptionSDDataLoaderConfig(ConfigBaseModel): class ImageCaptionFluxDataLoaderConfig (line 73) | class ImageCaptionFluxDataLoaderConfig(ConfigBaseModel): class DreamboothSDDataLoaderConfig (line 104) | class DreamboothSDDataLoaderConfig(ConfigBaseModel): class TextualInversionSDDataLoaderConfig (line 142) | class TextualInversionSDDataLoaderConfig(ConfigBaseModel): FILE: src/invoke_training/config/data/dataset_config.py class HFHubImageCaptionDatasetConfig (line 8) | class HFHubImageCaptionDatasetConfig(ConfigBaseModel): class ImageCaptionJsonlDatasetConfig (line 33) | class ImageCaptionJsonlDatasetConfig(ConfigBaseModel): class ImageDirDatasetConfig (line 54) | class ImageDirDatasetConfig(ConfigBaseModel): class ImageCaptionDirDatasetConfig (line 67) | class ImageCaptionDirDatasetConfig(ConfigBaseModel): FILE: src/invoke_training/config/optimizer/optimizer_config.py class AdamOptimizerConfig (line 6) | class AdamOptimizerConfig(ConfigBaseModel): class ProdigyOptimizerConfig (line 26) | class ProdigyOptimizerConfig(ConfigBaseModel): FILE: src/invoke_training/model_merge/extract_lora.py function get_patched_base_weights_from_peft_model (line 10) | def get_patched_base_weights_from_peft_model(peft_model: PeftModel) -> d... function get_state_dict_diff (line 29) | def get_state_dict_diff( function extract_lora_from_diffs (line 37) | def extract_lora_from_diffs( FILE: src/invoke_training/model_merge/merge_models.py function merge_models (line 10) | def merge_models( function lerp (line 53) | def lerp(a: torch.Tensor, b: torch.Tensor, weight_a: float) -> torch.Ten... function slerp (line 58) | def slerp(a: torch.Tensor, b: torch.Tensor, weight_a: float, dot_product... FILE: src/invoke_training/model_merge/merge_tasks_to_base.py function merge_tasks_to_base_model (line 9) | def merge_tasks_to_base_model( FILE: src/invoke_training/model_merge/scripts/extract_lora_from_model_diff.py class StableDiffusionModel (line 38) | class StableDiffusionModel: method all_none (line 45) | def all_none(self) -> bool: function load_model (line 49) | def load_model( function str_to_device (line 111) | def str_to_device(device_str: Literal["cuda", "cpu"]) -> torch.device: function state_dict_to_device (line 120) | def state_dict_to_device(state_dict: dict[str, torch.Tensor], device: to... function extract_lora_from_submodel (line 124) | def extract_lora_from_submodel( function extract_lora (line 187) | def extract_lora( function main (line 257) | def main(): FILE: src/invoke_training/model_merge/scripts/merge_lora_into_model.py function to_invokeai_base_model_type (line 24) | def to_invokeai_base_model_type(model_type: PipelineVersionEnum): function merge_lora_into_sd_model (line 34) | def merge_lora_into_sd_model( function parse_lora_model_arg (line 102) | def parse_lora_model_arg(lora_model_arg: str) -> tuple[str, float]: function main (line 113) | def main(): FILE: src/invoke_training/model_merge/scripts/merge_models.py class MergeModel (line 16) | class MergeModel: function run_merge_models (line 22) | def run_merge_models( function parse_model_args (line 73) | def parse_model_args(models: list[str], weights: list[str]) -> list[Merg... function main (line 85) | def main(): FILE: src/invoke_training/model_merge/scripts/merge_task_models_to_base_model.py function run_merge_models (line 14) | def run_merge_models( function main (line 91) | def main(): FILE: src/invoke_training/model_merge/utils/normalize_weights.py function normalize_weights (line 1) | def normalize_weights(weights: list[float]) -> list[float]: FILE: src/invoke_training/model_merge/utils/parse_model_arg.py function parse_model_arg (line 1) | def parse_model_arg(model: str, delimiter: str = "::") -> tuple[str, str... FILE: src/invoke_training/pipelines/_experimental/sd_dpo_lora/config.py class HFHubImagePairPreferenceDatasetConfig (line 10) | class HFHubImagePairPreferenceDatasetConfig(ConfigBaseModel): class ImagePairPreferenceDatasetConfig (line 16) | class ImagePairPreferenceDatasetConfig(ConfigBaseModel): class ImagePairPreferenceSDDataLoaderConfig (line 23) | class ImagePairPreferenceSDDataLoaderConfig(ConfigBaseModel): class SdDirectPreferenceOptimizationLoraConfig (line 50) | class SdDirectPreferenceOptimizationLoraConfig(BasePipelineConfig): method check_validation_prompts (line 242) | def check_validation_prompts(self): FILE: src/invoke_training/pipelines/_experimental/sd_dpo_lora/train.py function _save_sd_lora_checkpoint (line 47) | def _save_sd_lora_checkpoint( function train_forward_dpo (line 70) | def train_forward_dpo( # noqa: C901 function train (line 191) | def train(config: SdDirectPreferenceOptimizationLoraConfig, callbacks: l... FILE: src/invoke_training/pipelines/callbacks.py class ModelType (line 5) | class ModelType(Enum): class ModelCheckpoint (line 37) | class ModelCheckpoint: method __init__ (line 40) | def __init__(self, file_path: str, model_type: ModelType): class TrainingCheckpoint (line 45) | class TrainingCheckpoint: method __init__ (line 50) | def __init__(self, models: list[ModelCheckpoint], epoch: int, step: int): class ValidationImage (line 56) | class ValidationImage: method __init__ (line 57) | def __init__(self, file_path: str, prompt: str, image_idx: int): class ValidationImages (line 71) | class ValidationImages: method __init__ (line 72) | def __init__(self, images: list[ValidationImage], epoch: int, step: int): class PipelineCallbacks (line 85) | class PipelineCallbacks(ABC): method on_save_checkpoint (line 86) | def on_save_checkpoint(self, checkpoint: TrainingCheckpoint): method on_save_validation_images (line 89) | def on_save_validation_images(self, images: ValidationImages): FILE: src/invoke_training/pipelines/flux/lora/config.py class FluxLoraConfig (line 17) | class FluxLoraConfig(BasePipelineConfig): FILE: src/invoke_training/pipelines/flux/lora/train.py function _save_flux_lora_checkpoint (line 47) | def _save_flux_lora_checkpoint( function _build_data_loader (line 86) | def _build_data_loader( function cache_text_encoder_outputs (line 109) | def cache_text_encoder_outputs( function cache_vae_outputs (line 137) | def cache_vae_outputs(cache_dir: str, data_loader: DataLoader, vae: Auto... function get_sigmas (line 156) | def get_sigmas(noise_scheduler, timesteps, device, n_dim=4, dtype=torch.... function get_noisy_latents (line 168) | def get_noisy_latents(noise_scheduler: FlowMatchEulerDiscreteScheduler, ... function decode_latents (line 209) | def decode_latents(vae: AutoencoderKL, latents: torch.Tensor): function train_forward (line 222) | def train_forward( # noqa: C901 function train (line 304) | def train(config: FluxLoraConfig, callbacks: list[PipelineCallbacks] | N... FILE: src/invoke_training/pipelines/invoke_train.py function train (line 17) | def train(config: PipelineConfig, callbacks: list[PipelineCallbacks] | N... FILE: src/invoke_training/pipelines/stable_diffusion/lora/config.py class SdLoraConfig (line 14) | class SdLoraConfig(BasePipelineConfig): method check_validation_prompts (line 224) | def check_validation_prompts(self): FILE: src/invoke_training/pipelines/stable_diffusion/lora/train.py function _save_sd_lora_checkpoint (line 46) | def _save_sd_lora_checkpoint( function _build_data_loader (line 80) | def _build_data_loader( function cache_text_encoder_outputs (line 115) | def cache_text_encoder_outputs( function cache_vae_outputs (line 143) | def cache_vae_outputs(cache_dir: str, data_loader: DataLoader, vae: Auto... function train_forward (line 162) | def train_forward( # noqa: C901 function train (line 267) | def train(config: SdLoraConfig, callbacks: list[PipelineCallbacks] | Non... FILE: src/invoke_training/pipelines/stable_diffusion/textual_inversion/config.py class SdTextualInversionConfig (line 10) | class SdTextualInversionConfig(BasePipelineConfig): method check_validation_prompts (line 198) | def check_validation_prompts(self): FILE: src/invoke_training/pipelines/stable_diffusion/textual_inversion/train.py function _save_ti_embeddings (line 41) | def _save_ti_embeddings( function _initialize_placeholder_tokens (line 80) | def _initialize_placeholder_tokens( function train (line 138) | def train(config: SdTextualInversionConfig, callbacks: list[PipelineCall... FILE: src/invoke_training/pipelines/stable_diffusion_xl/finetune/config.py class SdxlFinetuneConfig (line 10) | class SdxlFinetuneConfig(BasePipelineConfig): method check_validation_prompts (line 166) | def check_validation_prompts(self): FILE: src/invoke_training/pipelines/stable_diffusion_xl/finetune/train.py function _save_sdxl_checkpoint (line 44) | def _save_sdxl_checkpoint( function train (line 96) | def train(config: SdxlFinetuneConfig, callbacks: list[PipelineCallbacks]... FILE: src/invoke_training/pipelines/stable_diffusion_xl/lora/config.py class SdxlLoraConfig (line 14) | class SdxlLoraConfig(BasePipelineConfig): method check_validation_prompts (line 230) | def check_validation_prompts(self): FILE: src/invoke_training/pipelines/stable_diffusion_xl/lora/train.py function _save_sdxl_lora_checkpoint (line 49) | def _save_sdxl_lora_checkpoint( function _build_data_loader (line 88) | def _build_data_loader( function _encode_prompt (line 131) | def _encode_prompt(text_encoders: list[CLIPPreTrainedModel], prompt_toke... function cache_text_encoder_outputs (line 158) | def cache_text_encoder_outputs( function train_forward (line 200) | def train_forward( # noqa: C901 function train (line 335) | def train(config: SdxlLoraConfig, callbacks: list[PipelineCallbacks] | N... FILE: src/invoke_training/pipelines/stable_diffusion_xl/lora_and_textual_inversion/config.py class SdxlLoraAndTextualInversionConfig (line 10) | class SdxlLoraAndTextualInversionConfig(BasePipelineConfig): method check_validation_prompts (line 233) | def check_validation_prompts(self): FILE: src/invoke_training/pipelines/stable_diffusion_xl/lora_and_textual_inversion/train.py function _save_sdxl_lora_and_ti_checkpoint (line 50) | def _save_sdxl_lora_and_ti_checkpoint( function train (line 118) | def train(config: SdxlLoraAndTextualInversionConfig, callbacks: list[Pip... FILE: src/invoke_training/pipelines/stable_diffusion_xl/textual_inversion/config.py class SdxlTextualInversionConfig (line 10) | class SdxlTextualInversionConfig(BasePipelineConfig): method check_validation_prompts (line 200) | def check_validation_prompts(self): FILE: src/invoke_training/pipelines/stable_diffusion_xl/textual_inversion/train.py function _save_ti_embeddings (line 44) | def _save_ti_embeddings( function _initialize_placeholder_tokens (line 93) | def _initialize_placeholder_tokens( function train (line 164) | def train(config: SdxlTextualInversionConfig, callbacks: list[PipelineCa... FILE: src/invoke_training/scripts/_experimental/auto_caption/auto_caption_images.py function select_device_and_dtype (line 14) | def select_device_and_dtype(force_cpu: bool = False) -> tuple[torch.devi... function process_images (line 24) | def process_images(images: list[Image.Image], prompt: str, moondream, to... function main (line 35) | def main( FILE: src/invoke_training/scripts/_experimental/masks/clipseg.py function load_clipseg_model (line 6) | def load_clipseg_model() -> tuple[CLIPSegProcessor, CLIPSegForImageSegme... function run_clipseg (line 13) | def run_clipseg( function select_device (line 58) | def select_device() -> torch.device: FILE: src/invoke_training/scripts/_experimental/masks/generate_masks.py function generate_masks (line 13) | def generate_masks(image_dir: str, prompt: str, clipseg_temp: float, bat... function main (line 53) | def main(): FILE: src/invoke_training/scripts/_experimental/masks/generate_masks_for_jsonl_dataset.py function collate_fn (line 16) | def collate_fn(examples): function validate_out_json_path (line 24) | def validate_out_json_path(out_json_path: str | Path): function generate_masks (line 33) | def generate_masks( function main (line 92) | def main(): FILE: src/invoke_training/scripts/_experimental/rank_images.py function parse_args (line 15) | def parse_args(): function clip (line 28) | def clip(val, min_val, max_val): function main (line 32) | def main(): FILE: src/invoke_training/scripts/convert_sd_lora_to_kohya_format.py function parse_args (line 11) | def parse_args(): function main (line 36) | def main(): FILE: src/invoke_training/scripts/invoke_generate_images.py function parse_args (line 8) | def parse_args(): function parse_lora_args (line 95) | def parse_lora_args(lora_args: list[str] | None) -> list[tuple[Path, int]]: function parse_prompt_file (line 113) | def parse_prompt_file(prompt_file: str) -> list[str]: function main (line 120) | def main(): FILE: src/invoke_training/scripts/invoke_train.py function parse_args (line 11) | def parse_args(): function main (line 23) | def main(): FILE: src/invoke_training/scripts/invoke_train_ui.py function main (line 8) | def main(): FILE: src/invoke_training/scripts/invoke_visualize_data_loading.py function save_image (line 26) | def save_image(torch_image: torch.Tensor, out_path: Path): function parse_args (line 49) | def parse_args(): function visualize (line 62) | def visualize(data_loader: DataLoader): function main (line 89) | def main(): FILE: src/invoke_training/scripts/utils/image_dir_dataset.py class ImageDirDataset (line 8) | class ImageDirDataset(torch.utils.data.Dataset): method __init__ (line 11) | def __init__( method _load_image (line 29) | def _load_image(self, image_path: str) -> Image.Image: method __len__ (line 34) | def __len__(self) -> int: method __getitem__ (line 37) | def __getitem__(self, idx: int): function list_collate_fn (line 43) | def list_collate_fn(examples): FILE: src/invoke_training/ui/app.py function build_app (line 12) | def build_app(): FILE: src/invoke_training/ui/config_groups/aspect_ratio_bucket_config_group.py class AspectRatioBucketConfigGroup (line 9) | class AspectRatioBucketConfigGroup(UIConfigElement): method __init__ (line 10) | def __init__(self): method update_ui_components_with_config_data (line 23) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 41) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/base_pipeline_config_group.py class BasePipelineConfigGroup (line 10) | class BasePipelineConfigGroup(UIConfigElement): method __init__ (line 11) | def __init__(self): method update_ui_components_with_config_data (line 55) | def update_ui_components_with_config_data(self, config: BasePipelineCo... method update_config_with_ui_component_data (line 94) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/dataset_config_group.py class HFHubImageCaptionDatasetConfigGroup (line 22) | class HFHubImageCaptionDatasetConfigGroup(UIConfigElement): method __init__ (line 23) | def __init__(self): method update_ui_components_with_config_data (line 43) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 54) | def update_config_with_ui_component_data( class ImageCaptionJsonlDatasetConfigGroup (line 70) | class ImageCaptionJsonlDatasetConfigGroup(UIConfigElement): method __init__ (line 71) | def __init__(self): method update_ui_components_with_config_data (line 90) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 104) | def update_config_with_ui_component_data( class ImageCaptionDirDatasetConfigGroup (line 119) | class ImageCaptionDirDatasetConfigGroup(UIConfigElement): method __init__ (line 120) | def __init__(self): method update_ui_components_with_config_data (line 133) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 141) | def update_config_with_ui_component_data( class ImageDirDatasetConfigGroup (line 153) | class ImageDirDatasetConfigGroup(UIConfigElement): method __init__ (line 154) | def __init__(self): method update_ui_components_with_config_data (line 167) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 175) | def update_config_with_ui_component_data( class DatasetConfigGroup (line 187) | class DatasetConfigGroup(UIConfigElement): method __init__ (line 188) | def __init__(self, allowed_types: list[str]): method _on_type_change (line 224) | def _on_type_change(self, type: str): method update_ui_components_with_config_data (line 232) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 270) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/flux_lora_config_group.py class FluxLoraConfigGroup (line 15) | class FluxLoraConfigGroup(UIConfigElement): method __init__ (line 16) | def __init__(self): method get_ui_output_components (line 185) | def get_ui_output_components(self) -> list[gr.components.Component]: method update_ui_components_with_config_data (line 224) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 285) | def update_config_with_ui_component_data( # noqa: C901 FILE: src/invoke_training/ui/config_groups/image_caption_sd_data_loader_config_group.py class ImageCaptionSDDataLoaderConfigGroup (line 11) | class ImageCaptionSDDataLoaderConfigGroup(UIConfigElement): method __init__ (line 12) | def __init__(self): method update_ui_components_with_config_data (line 63) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 81) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/optimizer_config_group.py class AdamOptimizerConfigGroup (line 11) | class AdamOptimizerConfigGroup(UIConfigElement): method __init__ (line 12) | def __init__(self): method update_ui_components_with_config_data (line 34) | def update_ui_components_with_config_data(self, config: AdamOptimizerC... method update_config_with_ui_component_data (line 44) | def update_config_with_ui_component_data( class ProdigyOptimizerConfigGroup (line 59) | class ProdigyOptimizerConfigGroup(UIConfigElement): method __init__ (line 60) | def __init__(self): method update_ui_components_with_config_data (line 77) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 87) | def update_config_with_ui_component_data( class OptimizerConfigGroup (line 100) | class OptimizerConfigGroup(UIConfigElement): method __init__ (line 101) | def __init__(self): method _on_optimizer_type_change (line 119) | def _on_optimizer_type_change(self, optimizer_type: str): method update_ui_components_with_config_data (line 125) | def update_ui_components_with_config_data(self, config: OptimizerConfi... method update_config_with_ui_component_data (line 145) | def update_config_with_ui_component_data(self, orig_config: OptimizerC... FILE: src/invoke_training/ui/config_groups/sd_lora_config_group.py class SdLoraConfigGroup (line 19) | class SdLoraConfigGroup(UIConfigElement): method __init__ (line 20) | def __init__(self): method update_ui_components_with_config_data (line 180) | def update_ui_components_with_config_data(self, config: SdLoraConfig) ... method update_config_with_ui_component_data (line 218) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/sd_textual_inversion_config_group.py class SdTextualInversionConfigGroup (line 19) | class SdTextualInversionConfigGroup(UIConfigElement): method __init__ (line 20) | def __init__(self): method update_ui_components_with_config_data (line 180) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 218) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/sdxl_finetune_config_group.py class SdxlFinetuneConfigGroup (line 19) | class SdxlFinetuneConfigGroup(UIConfigElement): method __init__ (line 20) | def __init__(self): method update_ui_components_with_config_data (line 177) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 215) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/sdxl_lora_and_textual_inversion_config_group.py class SdxlLoraAndTextualInversionConfigGroup (line 21) | class SdxlLoraAndTextualInversionConfigGroup(UIConfigElement): method __init__ (line 22) | def __init__(self): method update_ui_components_with_config_data (line 225) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 273) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/sdxl_lora_config_group.py class SdxlLoraConfigGroup (line 19) | class SdxlLoraConfigGroup(UIConfigElement): method __init__ (line 20) | def __init__(self): method update_ui_components_with_config_data (line 186) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 227) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/sdxl_textual_inversion_config_group.py class SdxlTextualInversionConfigGroup (line 19) | class SdxlTextualInversionConfigGroup(UIConfigElement): method __init__ (line 20) | def __init__(self): method update_ui_components_with_config_data (line 186) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 225) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/textual_inversion_sd_data_loader_config_group.py class TextualInversionSDDataLoaderConfigGroup (line 13) | class TextualInversionSDDataLoaderConfigGroup(UIConfigElement): method __init__ (line 14) | def __init__(self): method update_ui_components_with_config_data (line 88) | def update_ui_components_with_config_data( method update_config_with_ui_component_data (line 114) | def update_config_with_ui_component_data( FILE: src/invoke_training/ui/config_groups/ui_config_element.py class UIConfigElement (line 6) | class UIConfigElement: method get_ui_output_components (line 9) | def get_ui_output_components(self) -> list[gr.components.Component]: method get_ui_input_components (line 19) | def get_ui_input_components(self) -> list[gr.components.Component]: method update_ui_components_with_config_data (line 29) | def update_ui_components_with_config_data(self, config) -> dict[gr.com... method update_config_with_ui_component_data (line 33) | def update_config_with_ui_component_data(self, orig_config, ui_data: d... FILE: src/invoke_training/ui/gradio_blocks/header.py class Header (line 6) | class Header: method __init__ (line 7) | def __init__(self): FILE: src/invoke_training/ui/gradio_blocks/pipeline_tab.py class PipelineTab (line 11) | class PipelineTab: method __init__ (line 12) | def __init__( method on_reset_config_button_click (line 102) | def on_reset_config_button_click(self, file_path: str): method on_generate_config_button_click (line 129) | def on_generate_config_button_click(self, data: dict): method on_run_training_button_click (line 162) | def on_run_training_button_click(self): FILE: src/invoke_training/ui/pages/data_page.py class DataPage (line 18) | class DataPage: method __init__ (line 19) | def __init__(self): method _update_state (line 190) | def _update_state(self, idx: int): method _on_load_existing_dataset_button_click (line 234) | def _on_load_existing_dataset_button_click(self, data: dict): method _on_create_dataset_button_click (line 249) | def _on_create_dataset_button_click(self, data: dict): method _on_change_dataset_button_click (line 269) | def _on_change_dataset_button_click(self): method _on_save_and_go_button_click (line 274) | def _on_save_and_go_button_click(self, data: dict, idx_change: int): method _on_save_and_next_button_click (line 288) | def _on_save_and_next_button_click(self, data: dict): method _on_save_and_prev_button_click (line 291) | def _on_save_and_prev_button_click(self, data: dict): method _on_cur_example_index_change (line 294) | def _on_cur_example_index_change(self, data: dict): method _on_add_images_button_click (line 297) | def _on_add_images_button_click(self, data: dict): method app (line 336) | def app(self): FILE: src/invoke_training/ui/pages/training_page.py class TrainingPage (line 33) | class TrainingPage: method __init__ (line 34) | def __init__(self): method app (line 152) | def app(self): method _run_training (line 155) | def _run_training(self, config: PipelineConfig): FILE: src/invoke_training/ui/utils/prompts.py function split_pos_neg_prompts (line 4) | def split_pos_neg_prompts(prompt: str) -> tuple[str, str]: function merge_pos_neg_prompts (line 28) | def merge_pos_neg_prompts(positive_prompt: str, negative_prompt: str) ->... function convert_ui_prompts_to_pos_neg_prompts (line 47) | def convert_ui_prompts_to_pos_neg_prompts(prompts: str) -> tuple[list[st... function convert_pos_neg_prompts_to_ui_prompts (line 69) | def convert_pos_neg_prompts_to_ui_prompts(positive_prompts: list[str], n... FILE: src/invoke_training/ui/utils/utils.py function get_config_dir_path (line 10) | def get_config_dir_path() -> Path: function get_assets_dir_path (line 17) | def get_assets_dir_path() -> Path: function load_config_from_yaml (line 24) | def load_config_from_yaml(file_path: Path | str) -> PipelineConfig: function get_typing_literal_options (line 35) | def get_typing_literal_options(cls, field_name: str) -> list[str]: FILE: tests/invoke_training/_shared/checkpoints/test_checkpoint_tracker.py function test_checkpoint_tracker_get_path_file (line 10) | def test_checkpoint_tracker_get_path_file(): function test_checkpoint_tracker_get_path_directory (line 24) | def test_checkpoint_tracker_get_path_directory(): function test_checkpoint_tracker_bad_extension (line 38) | def test_checkpoint_tracker_bad_extension(): function test_checkpoint_tracker_prune_files (line 46) | def test_checkpoint_tracker_prune_files(): function test_checkpoint_tracker_prune_directories (line 65) | def test_checkpoint_tracker_prune_directories(): function test_checkpoint_tracker_prune_no_max (line 86) | def test_checkpoint_tracker_prune_no_max(): FILE: tests/invoke_training/_shared/checkpoints/test_serialization.py function test_state_dict_save_and_load_roundtrip (line 14) | def test_state_dict_save_and_load_roundtrip(file_name): function test_save_state_dict_bad_extension (line 29) | def test_save_state_dict_bad_extension(): function test_load_state_dict_bad_extension (line 35) | def test_load_state_dict_bad_extension(): FILE: tests/invoke_training/_shared/data/data_loaders/test_dreambooth_sd_dataloader.py function test_build_dreambooth_sd_dataloader (line 12) | def test_build_dreambooth_sd_dataloader(image_dir): # noqa: F811 function test_build_dreambooth_sd_dataloader_no_class_dataset (line 48) | def test_build_dreambooth_sd_dataloader_no_class_dataset(image_dir): # ... function test_build_dreambooth_sd_dataloader_with_bucketing (line 82) | def test_build_dreambooth_sd_dataloader_with_bucketing(image_dir): # no... FILE: tests/invoke_training/_shared/data/data_loaders/test_image_caption_sd_dataloader.py function test_build_image_caption_sd_dataloader (line 12) | def test_build_image_caption_sd_dataloader(image_caption_jsonl): # noqa... function test_build_image_caption_sd_dataloader_with_masks (line 41) | def test_build_image_caption_sd_dataloader_with_masks(image_caption_json... FILE: tests/invoke_training/_shared/data/data_loaders/test_image_pair_preference_sd_dataloader.py function test_build_image_pair_preference_sd_dataloader (line 16) | def test_build_image_pair_preference_sd_dataloader(): FILE: tests/invoke_training/_shared/data/data_loaders/test_textual_inversion_sd_dataloader.py function test_build_textual_inversion_sd_dataloader (line 15) | def test_build_textual_inversion_sd_dataloader(image_dir): # noqa: F811 function test_build_textual_inversion_sd_dataloader_keep_original_captions (line 50) | def test_build_textual_inversion_sd_dataloader_keep_original_captions(im... function test_build_textual_inversion_sd_dataloader_with_masks (line 72) | def test_build_textual_inversion_sd_dataloader_with_masks(image_caption_... FILE: tests/invoke_training/_shared/data/dataset_fixtures.py function image_dir (line 10) | def image_dir(tmp_path_factory: pytest.TempPathFactory): function image_caption_dir (line 30) | def image_caption_dir(tmp_path_factory: pytest.TempPathFactory): function image_caption_jsonl (line 53) | def image_caption_jsonl(tmp_path_factory: pytest.TempPathFactory): function image_pair_preference_dir (line 88) | def image_pair_preference_dir(tmp_path_factory: pytest.TempPathFactory): FILE: tests/invoke_training/_shared/data/datasets/test_hf_image_caption_dataset.py function create_hf_imagefolder_dataset (line 19) | def create_hf_imagefolder_dataset(tmp_dir: Path, num_images: int): function hf_imagefolder_dir (line 46) | def hf_imagefolder_dir(tmp_path_factory: pytest.TempPathFactory): function hf_dir_dataset (line 62) | def hf_dir_dataset(hf_imagefolder_dir: Path): function test_hf_dir_image_caption_dataset_bad_image_column (line 66) | def test_hf_dir_image_caption_dataset_bad_image_column(hf_imagefolder_di... function test_hf_dir_image_caption_dataset_bad_caption_column (line 74) | def test_hf_dir_image_caption_dataset_bad_caption_column(hf_imagefolder_... function test_hf_dir_image_caption_dataset_len (line 82) | def test_hf_dir_image_caption_dataset_len(hf_dir_dataset: HFImageCaption... function test_hf_dir_image_caption_dataset_index_error (line 87) | def test_hf_dir_image_caption_dataset_index_error(hf_dir_dataset: HFImag... function test_hf_dir_image_caption_dataset_getitem (line 93) | def test_hf_dir_image_caption_dataset_getitem(hf_dir_dataset: HFImageCap... function test_hf_dir_image_caption_dataset_get_image_dimensions (line 104) | def test_hf_dir_image_caption_dataset_get_image_dimensions(hf_dir_datase... function test_hf_hub_image_caption_dataset_bad_image_column (line 121) | def test_hf_hub_image_caption_dataset_bad_image_column(): function test_hf_hub_image_caption_dataset_bad_caption_column (line 135) | def test_hf_hub_image_caption_dataset_bad_caption_column(): function hf_hub_dataset (line 148) | def hf_hub_dataset(): function test_hf_hub_image_caption_dataset_index_error (line 157) | def test_hf_hub_image_caption_dataset_index_error(hf_hub_dataset: HFImag... function test_hf_hub_image_caption_dataset_len (line 165) | def test_hf_hub_image_caption_dataset_len(hf_hub_dataset: HFImageCaption... function test_hf_hub_image_caption_dataset_getitem (line 174) | def test_hf_hub_image_caption_dataset_getitem(hf_hub_dataset: HFImageCap... function test_hf_hub_image_caption_dataset_get_image_dimensions (line 187) | def test_hf_hub_image_caption_dataset_get_image_dimensions(hf_hub_datase... FILE: tests/invoke_training/_shared/data/datasets/test_hf_image_pair_preference_dataset.py function test_hf_hub_image_caption_dataset_getitem (line 9) | def test_hf_hub_image_caption_dataset_getitem(): function test_hf_hub_image_caption_dataset_len (line 45) | def test_hf_hub_image_caption_dataset_len(): function test_hf_hub_image_caption_dataset_skip_no_preference_len (line 66) | def test_hf_hub_image_caption_dataset_skip_no_preference_len(): FILE: tests/invoke_training/_shared/data/datasets/test_image_caption_dir_dataset.py function test_image_caption_dir_dataset_len (line 11) | def test_image_caption_dir_dataset_len(image_caption_dir): # noqa: F811 function test_image_caption_dir_dataset_getitem (line 17) | def test_image_caption_dir_dataset_getitem(image_caption_dir): # noqa: ... function test_image_caption_dir_dataset_keep_in_memory (line 29) | def test_image_caption_dir_dataset_keep_in_memory(image_caption_dir): #... function test_image_caption_dir_dataset_get_image_dimensions (line 41) | def test_image_caption_dir_dataset_get_image_dimensions(image_caption_di... function test_image_caption_dir_dataset_missing_caption_file (line 49) | def test_image_caption_dir_dataset_missing_caption_file(tmp_path: Path):... FILE: tests/invoke_training/_shared/data/datasets/test_image_caption_jsonl_dataset.py function test_image_caption_jsonl_dataset_len (line 12) | def test_image_caption_jsonl_dataset_len(image_caption_jsonl): # noqa: ... function test_image_caption_jsonl_dataset_getitem (line 18) | def test_image_caption_jsonl_dataset_getitem(image_caption_jsonl): # no... function test_image_caption_jsonl_dataset_keep_in_memory (line 32) | def test_image_caption_jsonl_dataset_keep_in_memory(image_caption_jsonl)... function test_image_caption_jsonl_dataset_get_image_dimensions (line 53) | def test_image_caption_jsonl_dataset_get_image_dimensions(image_caption_... function test_image_caption_jsonl_dataset_save_jsonl (line 61) | def test_image_caption_jsonl_dataset_save_jsonl(image_caption_jsonl, tmp... FILE: tests/invoke_training/_shared/data/datasets/test_image_dir_dataset.py function test_image_dir_dataset_len (line 8) | def test_image_dir_dataset_len(image_dir): # noqa: F811 function test_image_dir_dataset_getitem (line 14) | def test_image_dir_dataset_getitem(image_dir): # noqa: F811 function test_image_dir_dataset_keep_in_memory (line 25) | def test_image_dir_dataset_keep_in_memory(image_dir): # noqa: F811 function test_image_dir_dataset_get_image_dimensions (line 43) | def test_image_dir_dataset_get_image_dimensions(image_dir): # noqa: F811 FILE: tests/invoke_training/_shared/data/datasets/test_image_pair_preference_dataset.py function test_image_dir_dataset_len (line 8) | def test_image_dir_dataset_len(image_pair_preference_dir): # noqa: F811 function test_image_dir_dataset_getitem (line 14) | def test_image_dir_dataset_getitem(image_pair_preference_dir): # noqa: ... FILE: tests/invoke_training/_shared/data/datasets/test_transform_dataset.py function test_transform_dataset_len (line 6) | def test_transform_dataset_len(): function test_transform_dataset_getitem (line 16) | def test_transform_dataset_getitem(): FILE: tests/invoke_training/_shared/data/samplers/test_aspect_ratio_bucket_batch_sampler.py function assert_shuffled_samples_match (line 8) | def assert_shuffled_samples_match(samples_1, samples_2): function test_aspect_ratio_bucket_batch_sampler (line 18) | def test_aspect_ratio_bucket_batch_sampler(): function test_aspect_ratio_bucket_batch_sampler_len (line 30) | def test_aspect_ratio_bucket_batch_sampler_len(): function test_aspect_ratio_bucket_batch_sampler_from_image_sizes (line 42) | def test_aspect_ratio_bucket_batch_sampler_from_image_sizes(): function test_aspect_ratio_bucket_batch_sampler_shuffle (line 66) | def test_aspect_ratio_bucket_batch_sampler_shuffle(): function test_aspect_ratio_bucket_batch_sampler_seed (line 81) | def test_aspect_ratio_bucket_batch_sampler_seed(): FILE: tests/invoke_training/_shared/data/samplers/test_batch_offset_sampler.py function test_batch_offset_sampler (line 6) | def test_batch_offset_sampler(): function test_batch_offset_sampler_len (line 18) | def test_batch_offset_sampler_len(): FILE: tests/invoke_training/_shared/data/samplers/test_concat_sampler.py function test_concat_sampler (line 4) | def test_concat_sampler(): function test_concat_sampler_batches (line 16) | def test_concat_sampler_batches(): function test_concat_sampler_len (line 28) | def test_concat_sampler_len(): FILE: tests/invoke_training/_shared/data/samplers/test_interleaved_sampler.py function test_interleaved_sampler (line 4) | def test_interleaved_sampler(): function test_interleaved_sampler_batches (line 16) | def test_interleaved_sampler_batches(): function test_interleaved_sampler_len (line 28) | def test_interleaved_sampler_len(): FILE: tests/invoke_training/_shared/data/samplers/test_offset_sampler.py function test_offset_sampler (line 6) | def test_offset_sampler(): function test_offset_sampler_len (line 16) | def test_offset_sampler_len(): FILE: tests/invoke_training/_shared/data/transforms/test_caption_prefix_transform.py function test_caption_prefix_transform (line 4) | def test_caption_prefix_transform(): FILE: tests/invoke_training/_shared/data/transforms/test_concat_fields_transform.py function test_caption_prefix_transform (line 4) | def test_caption_prefix_transform(): FILE: tests/invoke_training/_shared/data/transforms/test_constant_field_transform.py function test_constant_field_transform (line 4) | def test_constant_field_transform(): FILE: tests/invoke_training/_shared/data/transforms/test_drop_field_transform.py function test_drop_field_transform (line 4) | def test_drop_field_transform(): FILE: tests/invoke_training/_shared/data/transforms/test_load_cache_transform.py function test_load_cache_transform (line 8) | def test_load_cache_transform(): FILE: tests/invoke_training/_shared/data/transforms/test_sd_image_transform.py function denormalize_image (line 13) | def denormalize_image(img: np.ndarray) -> np.ndarray: function denormalize_mask (line 32) | def denormalize_mask(mask: np.ndarray) -> np.ndarray: function test_sd_image_transform_resolution (line 41) | def test_sd_image_transform_resolution(): function test_sd_image_transform_without_mask (line 69) | def test_sd_image_transform_without_mask(): function test_sd_image_transform_range (line 92) | def test_sd_image_transform_range(): function test_sd_image_transform_center_crop (line 124) | def test_sd_image_transform_center_crop(): function test_sd_image_transform_random_crop (line 155) | def test_sd_image_transform_random_crop(): function test_sd_image_transform_center_crop_flip (line 193) | def test_sd_image_transform_center_crop_flip(): function test_sd_image_transform_random_crop_flip (line 228) | def test_sd_image_transform_random_crop_flip(): function test_sd_image_transform_aspect_ratio_bucket_manager (line 270) | def test_sd_image_transform_aspect_ratio_bucket_manager(): function test_sd_image_transform_resolution_input_validation (line 309) | def test_sd_image_transform_resolution_input_validation( FILE: tests/invoke_training/_shared/data/transforms/test_shuffle_caption_transform.py function test_shuffle_caption_transform (line 4) | def test_shuffle_caption_transform(): function test_shuffle_caption_transform_no_delimiter (line 15) | def test_shuffle_caption_transform_no_delimiter(): FILE: tests/invoke_training/_shared/data/transforms/test_template_caption_transform.py function test_template_caption_transform (line 8) | def test_template_caption_transform(): function test_template_caption_transform_seed (line 20) | def test_template_caption_transform_seed(): function test_template_caption_transform_bad_templates (line 55) | def test_template_caption_transform_bad_templates(): FILE: tests/invoke_training/_shared/data/transforms/test_tensor_disk_cache.py function test_tensor_disk_cache_roundtrip (line 9) | def test_tensor_disk_cache_roundtrip(tmp_path: Path): function test_tensor_disk_cache_fail_overwrite (line 26) | def test_tensor_disk_cache_fail_overwrite(tmp_path): FILE: tests/invoke_training/_shared/data/utils/test_aspect_ratio_bucket_manager.py function test_build_aspect_ratio_buckets_input_validation (line 19) | def test_build_aspect_ratio_buckets_input_validation( function test_build_aspect_ratio_buckets (line 54) | def test_build_aspect_ratio_buckets( function test_get_aspect_ratio_bucket (line 79) | def test_get_aspect_ratio_bucket(resolution: Resolution, expected_bucket... FILE: tests/invoke_training/_shared/data/utils/test_resize.py function test_resize_to_cover (line 28) | def test_resize_to_cover(in_resolution: Resolution, size_to_cover: Resol... FILE: tests/invoke_training/_shared/data/utils/test_resolution.py function test_resolution_parse (line 14) | def test_resolution_parse(input, expected_resolution: Resolution): FILE: tests/invoke_training/_shared/stable_diffusion/test_base_model_version.py function test_get_base_model_version (line 21) | def test_get_base_model_version(diffusers_model_name: str, expected_vers... function test_check_base_model_version_pass (line 39) | def test_check_base_model_version_pass(): function test_check_base_model_version_fail (line 45) | def test_check_base_model_version_fail(): FILE: tests/invoke_training/_shared/stable_diffusion/test_lora_checkpoint_utils.py function test_convert_sd_peft_checkpoint_to_kohya_state_dict_raise_on_empty_directory (line 10) | def test_convert_sd_peft_checkpoint_to_kohya_state_dict_raise_on_empty_d... function test_convert_sd_peft_checkpoint_to_kohya_state_dict_raise_on_unexpected_subdirectory (line 17) | def test_convert_sd_peft_checkpoint_to_kohya_state_dict_raise_on_unexpec... FILE: tests/invoke_training/_shared/stable_diffusion/test_model_loading_utils.py function test_load_models_sd (line 17) | def test_load_models_sd(sdv1_embedding_path): # noqa: F811 function test_load_models_sdxl (line 38) | def test_load_models_sdxl(sdxl_embedding_path: Path): # noqa: F811 FILE: tests/invoke_training/_shared/stable_diffusion/test_textual_inversion.py function test_expand_placeholder_token (line 22) | def test_expand_placeholder_token(placeholder_token: str, num_vectors: i... function test_expand_placeholder_token_raises_on_invalid_num_vectors (line 26) | def test_expand_placeholder_token_raises_on_invalid_num_vectors(): function test_initialize_placeholder_tokens_from_initializer_token (line 32) | def test_initialize_placeholder_tokens_from_initializer_token(): function test_initialize_placeholder_tokens_from_initial_phrase (line 60) | def test_initialize_placeholder_tokens_from_initial_phrase(): function test_initialize_placeholder_tokens_from_initial_embedding (line 87) | def test_initialize_placeholder_tokens_from_initial_embedding(sdv1_embed... FILE: tests/invoke_training/_shared/stable_diffusion/ti_embedding_checkpoint_fixture.py function sdv1_embedding_path (line 8) | def sdv1_embedding_path(tmp_path_factory: pytest.TempPathFactory): function sdxl_embedding_path (line 26) | def sdxl_embedding_path(tmp_path_factory: pytest.TempPathFactory): FILE: tests/invoke_training/_shared/utils/test_jsonl.py function test_jsonl_roundtrip (line 6) | def test_jsonl_roundtrip(tmp_path: Path): FILE: tests/invoke_training/config/pipelines/test_pipeline_config.py function test_pipeline_config (line 10) | def test_pipeline_config(): FILE: tests/invoke_training/model_merge/test_merge_models.py function test_merge_models_raises_on_not_enough_state_dicts (line 12) | def test_merge_models_raises_on_not_enough_state_dicts(): function test_merge_models_raises_on_mismatched_weights (line 17) | def test_merge_models_raises_on_mismatched_weights(): function test_merge_models (line 88) | def test_merge_models( FILE: tests/invoke_training/model_merge/test_merge_tasks_to_base.py function test_merge_raises_on_mismatched_weights (line 11) | def test_merge_raises_on_mismatched_weights(): function test_merge_ties (line 65) | def test_merge_ties( FILE: tests/invoke_training/model_merge/utils.py function state_dicts_are_close (line 4) | def state_dicts_are_close(a: dict[str, torch.Tensor], b: dict[str, torch... FILE: tests/invoke_training/ui/utils/test_prompts.py function test_split_pos_neg_prompts (line 21) | def test_split_pos_neg_prompts(prompt: str, expected_positive_prompt: st... function test_split_pos_neg_prompts_raises_value_error (line 34) | def test_split_pos_neg_prompts_raises_value_error(prompt: str): function test_convert_ui_prompts_to_pos_neg_prompts (line 73) | def test_convert_ui_prompts_to_pos_neg_prompts( function test_convert_pos_neg_prompts_to_ui_prompts (line 82) | def test_convert_pos_neg_prompts_to_ui_prompts(