LICENSE
README.md
pyproject.toml
ostris_ai_toolkit.egg-info/PKG-INFO
ostris_ai_toolkit.egg-info/SOURCES.txt
ostris_ai_toolkit.egg-info/dependency_links.txt
ostris_ai_toolkit.egg-info/requires.txt
ostris_ai_toolkit.egg-info/top_level.txt
src/ostris-ai-toolkit/__init__.py
src/ostris-ai-toolkit/flux_train_ui.py
src/ostris-ai-toolkit/info.py
src/ostris-ai-toolkit/run.py
src/ostris-ai-toolkit/run_modal.py
src/ostris-ai-toolkit/config/examples/extract.example.yml
src/ostris-ai-toolkit/config/examples/generate.example.yaml
src/ostris-ai-toolkit/config/examples/mod_lora_scale.yaml
src/ostris-ai-toolkit/config/examples/train_lora_flux_24gb.yaml
src/ostris-ai-toolkit/config/examples/train_lora_flux_schnell_24gb.yaml
src/ostris-ai-toolkit/config/examples/train_lora_sd35_large_24gb.yaml
src/ostris-ai-toolkit/config/examples/train_slider.example.yml
src/ostris-ai-toolkit/config/examples/modal/modal_train_lora_flux_24gb.yaml
src/ostris-ai-toolkit/config/examples/modal/modal_train_lora_flux_schnell_24gb.yaml
src/ostris-ai-toolkit/extensions/example/ExampleMergeModels.py
src/ostris-ai-toolkit/extensions/example/__init__.py
src/ostris-ai-toolkit/extensions/example/config/config.example.yaml
src/ostris-ai-toolkit/extensions_built_in/__init__.py
src/ostris-ai-toolkit/extensions_built_in/advanced_generator/Img2ImgGenerator.py
src/ostris-ai-toolkit/extensions_built_in/advanced_generator/PureLoraGenerator.py
src/ostris-ai-toolkit/extensions_built_in/advanced_generator/ReferenceGenerator.py
src/ostris-ai-toolkit/extensions_built_in/advanced_generator/__init__.py
src/ostris-ai-toolkit/extensions_built_in/advanced_generator/config/train.example.yaml
src/ostris-ai-toolkit/extensions_built_in/concept_replacer/ConceptReplacer.py
src/ostris-ai-toolkit/extensions_built_in/concept_replacer/__init__.py
src/ostris-ai-toolkit/extensions_built_in/concept_replacer/config/train.example.yaml
src/ostris-ai-toolkit/extensions_built_in/dataset_tools/DatasetTools.py
src/ostris-ai-toolkit/extensions_built_in/dataset_tools/SuperTagger.py
src/ostris-ai-toolkit/extensions_built_in/dataset_tools/SyncFromCollection.py
src/ostris-ai-toolkit/extensions_built_in/dataset_tools/__init__.py
src/ostris-ai-toolkit/extensions_built_in/dataset_tools/tools/caption.py
src/ostris-ai-toolkit/extensions_built_in/dataset_tools/tools/dataset_tools_config_modules.py
src/ostris-ai-toolkit/extensions_built_in/dataset_tools/tools/fuyu_utils.py
src/ostris-ai-toolkit/extensions_built_in/dataset_tools/tools/image_tools.py
src/ostris-ai-toolkit/extensions_built_in/dataset_tools/tools/llava_utils.py
src/ostris-ai-toolkit/extensions_built_in/dataset_tools/tools/sync_tools.py
src/ostris-ai-toolkit/extensions_built_in/image_reference_slider_trainer/ImageReferenceSliderTrainerProcess.py
src/ostris-ai-toolkit/extensions_built_in/image_reference_slider_trainer/__init__.py
src/ostris-ai-toolkit/extensions_built_in/image_reference_slider_trainer/config/train.example.yaml
src/ostris-ai-toolkit/extensions_built_in/sd_trainer/SDTrainer.py
src/ostris-ai-toolkit/extensions_built_in/sd_trainer/__init__.py
src/ostris-ai-toolkit/extensions_built_in/sd_trainer/config/train.example.yaml
src/ostris-ai-toolkit/extensions_built_in/ultimate_slider_trainer/UltimateSliderTrainerProcess.py
src/ostris-ai-toolkit/extensions_built_in/ultimate_slider_trainer/__init__.py
src/ostris-ai-toolkit/extensions_built_in/ultimate_slider_trainer/config/train.example.yaml
src/ostris-ai-toolkit/jobs/BaseJob.py
src/ostris-ai-toolkit/jobs/ExtensionJob.py
src/ostris-ai-toolkit/jobs/ExtractJob.py
src/ostris-ai-toolkit/jobs/GenerateJob.py
src/ostris-ai-toolkit/jobs/MergeJob.py
src/ostris-ai-toolkit/jobs/ModJob.py
src/ostris-ai-toolkit/jobs/TrainJob.py
src/ostris-ai-toolkit/jobs/__init__.py
src/ostris-ai-toolkit/jobs/process/BaseExtensionProcess.py
src/ostris-ai-toolkit/jobs/process/BaseExtractProcess.py
src/ostris-ai-toolkit/jobs/process/BaseMergeProcess.py
src/ostris-ai-toolkit/jobs/process/BaseProcess.py
src/ostris-ai-toolkit/jobs/process/BaseSDTrainProcess.py
src/ostris-ai-toolkit/jobs/process/BaseTrainProcess.py
src/ostris-ai-toolkit/jobs/process/ExtractLoconProcess.py
src/ostris-ai-toolkit/jobs/process/ExtractLoraProcess.py
src/ostris-ai-toolkit/jobs/process/GenerateProcess.py
src/ostris-ai-toolkit/jobs/process/MergeLoconProcess.py
src/ostris-ai-toolkit/jobs/process/ModRescaleLoraProcess.py
src/ostris-ai-toolkit/jobs/process/TrainESRGANProcess.py
src/ostris-ai-toolkit/jobs/process/TrainFineTuneProcess.py
src/ostris-ai-toolkit/jobs/process/TrainSDRescaleProcess.py
src/ostris-ai-toolkit/jobs/process/TrainSliderProcess.py
src/ostris-ai-toolkit/jobs/process/TrainSliderProcessOld.py
src/ostris-ai-toolkit/jobs/process/TrainVAEProcess.py
src/ostris-ai-toolkit/jobs/process/__init__.py
src/ostris-ai-toolkit/jobs/process/models/__init__.py
src/ostris-ai-toolkit/jobs/process/models/vgg19_critic.py
src/ostris-ai-toolkit/scripts/__init__.py
src/ostris-ai-toolkit/scripts/convert_cog.py
src/ostris-ai-toolkit/scripts/convert_lora_to_peft_format.py
src/ostris-ai-toolkit/scripts/generate_sampler_step_scales.py
src/ostris-ai-toolkit/scripts/make_diffusers_model.py
src/ostris-ai-toolkit/scripts/make_lcm_sdxl_model.py
src/ostris-ai-toolkit/scripts/patch_te_adapter.py
src/ostris-ai-toolkit/scripts/repair_dataset_folder.py
src/ostris-ai-toolkit/testing/compare_keys.py
src/ostris-ai-toolkit/testing/generate_lora_mapping.py
src/ostris-ai-toolkit/testing/generate_weight_mappings.py
src/ostris-ai-toolkit/testing/merge_in_text_encoder_adapter.py
src/ostris-ai-toolkit/testing/shrink_pixart.py
src/ostris-ai-toolkit/testing/shrink_pixart2.py
src/ostris-ai-toolkit/testing/shrink_pixart_sm.py
src/ostris-ai-toolkit/testing/shrink_pixart_sm2.py
src/ostris-ai-toolkit/testing/shrink_pixart_sm3.py
src/ostris-ai-toolkit/testing/test_bucket_dataloader.py
src/ostris-ai-toolkit/testing/test_model_load_save.py
src/ostris-ai-toolkit/testing/test_vae.py
src/ostris-ai-toolkit/testing/test_vae_cycle.py
src/ostris-ai-toolkit/toolkit/__init__.py
src/ostris-ai-toolkit/toolkit/assistant_lora.py
src/ostris-ai-toolkit/toolkit/basic.py
src/ostris-ai-toolkit/toolkit/buckets.py
src/ostris-ai-toolkit/toolkit/civitai.py
src/ostris-ai-toolkit/toolkit/clip_vision_adapter.py
src/ostris-ai-toolkit/toolkit/config.py
src/ostris-ai-toolkit/toolkit/config_modules.py
src/ostris-ai-toolkit/toolkit/cuda_malloc.py
src/ostris-ai-toolkit/toolkit/custom_adapter.py
src/ostris-ai-toolkit/toolkit/data_loader.py
src/ostris-ai-toolkit/toolkit/dataloader_mixins.py
src/ostris-ai-toolkit/toolkit/dequantize.py
src/ostris-ai-toolkit/toolkit/ema.py
src/ostris-ai-toolkit/toolkit/embedding.py
src/ostris-ai-toolkit/toolkit/esrgan_utils.py
src/ostris-ai-toolkit/toolkit/extension.py
src/ostris-ai-toolkit/toolkit/guidance.py
src/ostris-ai-toolkit/toolkit/image_utils.py
src/ostris-ai-toolkit/toolkit/inversion_utils.py
src/ostris-ai-toolkit/toolkit/ip_adapter.py
src/ostris-ai-toolkit/toolkit/job.py
src/ostris-ai-toolkit/toolkit/kohya_model_util.py
src/ostris-ai-toolkit/toolkit/layers.py
src/ostris-ai-toolkit/toolkit/llvae.py
src/ostris-ai-toolkit/toolkit/logging.py
src/ostris-ai-toolkit/toolkit/lora_special.py
src/ostris-ai-toolkit/toolkit/lorm.py
src/ostris-ai-toolkit/toolkit/losses.py
src/ostris-ai-toolkit/toolkit/lycoris_special.py
src/ostris-ai-toolkit/toolkit/lycoris_utils.py
src/ostris-ai-toolkit/toolkit/metadata.py
src/ostris-ai-toolkit/toolkit/network_mixins.py
src/ostris-ai-toolkit/toolkit/optimizer.py
src/ostris-ai-toolkit/toolkit/paths.py
src/ostris-ai-toolkit/toolkit/photomaker.py
src/ostris-ai-toolkit/toolkit/photomaker_pipeline.py
src/ostris-ai-toolkit/toolkit/pipelines.py
src/ostris-ai-toolkit/toolkit/progress_bar.py
src/ostris-ai-toolkit/toolkit/prompt_utils.py
src/ostris-ai-toolkit/toolkit/reference_adapter.py
src/ostris-ai-toolkit/toolkit/resampler.py
src/ostris-ai-toolkit/toolkit/sampler.py
src/ostris-ai-toolkit/toolkit/saving.py
src/ostris-ai-toolkit/toolkit/scheduler.py
src/ostris-ai-toolkit/toolkit/sd_device_states_presets.py
src/ostris-ai-toolkit/toolkit/stable_diffusion_model.py
src/ostris-ai-toolkit/toolkit/style.py
src/ostris-ai-toolkit/toolkit/timer.py
src/ostris-ai-toolkit/toolkit/train_pipelines.py
src/ostris-ai-toolkit/toolkit/train_tools.py
src/ostris-ai-toolkit/toolkit/data_transfer_object/__init__.py
src/ostris-ai-toolkit/toolkit/data_transfer_object/data_loader.py
src/ostris-ai-toolkit/toolkit/keymaps/__init__.py
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_refiner.json
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_refiner_ldm_base.safetensors
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_refiner_unmatched.json
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_sd1.json
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_sd1_ldm_base.safetensors
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_sd2.json
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_sd2_ldm_base.safetensors
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_sd2_unmatched.json
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_sdxl.json
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_sdxl_ldm_base.safetensors
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_sdxl_unmatched.json
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_ssd.json
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_ssd_ldm_base.safetensors
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_ssd_unmatched.json
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_vega.json
src/ostris-ai-toolkit/toolkit/keymaps/stable_diffusion_vega_ldm_base.safetensors
src/ostris-ai-toolkit/toolkit/models/DoRA.py
src/ostris-ai-toolkit/toolkit/models/LoRAFormer.py
src/ostris-ai-toolkit/toolkit/models/RRDB.py
src/ostris-ai-toolkit/toolkit/models/__init__.py
src/ostris-ai-toolkit/toolkit/models/auraflow.py
src/ostris-ai-toolkit/toolkit/models/block.py
src/ostris-ai-toolkit/toolkit/models/clip_fusion.py
src/ostris-ai-toolkit/toolkit/models/clip_pre_processor.py
src/ostris-ai-toolkit/toolkit/models/ilora.py
src/ostris-ai-toolkit/toolkit/models/ilora2.py
src/ostris-ai-toolkit/toolkit/models/pixtral_vision.py
src/ostris-ai-toolkit/toolkit/models/single_value_adapter.py
src/ostris-ai-toolkit/toolkit/models/size_agnostic_feature_encoder.py
src/ostris-ai-toolkit/toolkit/models/te_adapter.py
src/ostris-ai-toolkit/toolkit/models/te_aug_adapter.py
src/ostris-ai-toolkit/toolkit/models/vd_adapter.py
src/ostris-ai-toolkit/toolkit/models/zipper_resampler.py
src/ostris-ai-toolkit/toolkit/optimizers/adafactor.py
src/ostris-ai-toolkit/toolkit/optimizers/optimizer_utils.py
src/ostris-ai-toolkit/toolkit/optimizers/prodigy_8bit.py
src/ostris-ai-toolkit/toolkit/orig_configs/__init__.py
src/ostris-ai-toolkit/toolkit/orig_configs/sd_xl_refiner.yaml
src/ostris-ai-toolkit/toolkit/samplers/__init__.py
src/ostris-ai-toolkit/toolkit/samplers/custom_flowmatch_sampler.py
src/ostris-ai-toolkit/toolkit/samplers/custom_lcm_scheduler.py
src/ostris-ai-toolkit/toolkit/util/__init__.py
src/ostris-ai-toolkit/toolkit/util/adafactor_stochastic_rounding.py
src/ostris-ai-toolkit/toolkit/util/inverse_cfg.py