├── .github ├── FUNDING.yml └── ISSUE_TEMPLATE │ ├── bug_report.md │ └── config.yml ├── .gitignore ├── .gitmodules ├── .vscode └── launch.json ├── DEPLOYMENT_GUIDE.md ├── DOCKER_README.md ├── FAQ.md ├── LICENSE ├── README.md ├── assets ├── VAE_test1.jpg ├── glif.svg └── lora_ease_ui.png ├── config └── examples │ ├── extract.example.yml │ ├── generate.example.yaml │ ├── mod_lora_scale.yaml │ ├── modal │ ├── modal_train_lora_flux_24gb.yaml │ └── modal_train_lora_flux_schnell_24gb.yaml │ ├── train_flex_redux.yaml │ ├── train_full_fine_tune_flex.yaml │ ├── train_full_fine_tune_lumina.yaml │ ├── train_lora_chroma_24gb.yaml │ ├── train_lora_flex2_24gb.yaml │ ├── train_lora_flex_24gb.yaml │ ├── train_lora_flux_24gb.yaml │ ├── train_lora_flux_kontext_24gb.yaml │ ├── train_lora_flux_schnell_24gb.yaml │ ├── train_lora_hidream_48.yaml │ ├── train_lora_lumina.yaml │ ├── train_lora_omnigen2_24gb.yaml │ ├── train_lora_qwen_image_24gb.yaml │ ├── train_lora_qwen_image_edit_2509_32gb.yaml │ ├── train_lora_qwen_image_edit_32gb.yaml │ ├── train_lora_sd35_large_24gb.yaml │ ├── train_lora_wan21_14b_24gb.yaml │ ├── train_lora_wan21_1b_24gb.yaml │ ├── train_lora_wan22_14b_24gb.yaml │ ├── train_lora_zimage_24gb.yaml │ └── train_slider.example.yml ├── deploy.sh ├── docker-compose.yml ├── docker ├── Dockerfile └── start.sh ├── extensions └── example │ ├── ExampleMergeModels.py │ ├── __init__.py │ └── config │ └── config.example.yaml ├── extensions_built_in ├── advanced_generator │ ├── Img2ImgGenerator.py │ ├── PureLoraGenerator.py │ ├── ReferenceGenerator.py │ ├── __init__.py │ └── config │ │ └── train.example.yaml ├── concept_replacer │ ├── ConceptReplacer.py │ ├── __init__.py │ └── config │ │ └── train.example.yaml ├── concept_slider │ ├── ConceptSliderTrainer.py │ └── __init__.py ├── dataset_tools │ ├── DatasetTools.py │ ├── SuperTagger.py │ ├── SyncFromCollection.py │ ├── __init__.py │ └── tools │ │ ├── caption.py │ │ ├── dataset_tools_config_modules.py │ │ ├── fuyu_utils.py │ │ ├── image_tools.py │ │ ├── llava_utils.py │ │ └── sync_tools.py ├── diffusion_models │ ├── __init__.py │ ├── chroma │ │ ├── __init__.py │ │ ├── chroma_model.py │ │ ├── chroma_radiance_model.py │ │ ├── pipeline.py │ │ └── src │ │ │ ├── __init__.py │ │ │ ├── layers.py │ │ │ ├── math.py │ │ │ ├── model.py │ │ │ └── radiance.py │ ├── f_light │ │ ├── __init__.py │ │ ├── f_light.py │ │ └── src │ │ │ ├── __init__.py │ │ │ ├── model.py │ │ │ └── pipeline.py │ ├── flux2 │ │ ├── __init__.py │ │ ├── flux2_model.py │ │ └── src │ │ │ ├── __init__.py │ │ │ ├── autoencoder.py │ │ │ ├── model.py │ │ │ ├── pipeline.py │ │ │ └── sampling.py │ ├── flux_kontext │ │ ├── __init__.py │ │ └── flux_kontext.py │ ├── hidream │ │ ├── __init__.py │ │ ├── hidream_e1_model.py │ │ ├── hidream_model.py │ │ └── src │ │ │ ├── __init__.py │ │ │ ├── models │ │ │ ├── attention.py │ │ │ ├── attention_processor.py │ │ │ ├── embeddings.py │ │ │ ├── moe.py │ │ │ └── transformers │ │ │ │ └── transformer_hidream_image.py │ │ │ ├── pipelines │ │ │ └── hidream_image │ │ │ │ ├── pipeline_hidream_image.py │ │ │ │ ├── pipeline_hidream_image_editing.py │ │ │ │ └── pipeline_output.py │ │ │ └── schedulers │ │ │ ├── flash_flow_match.py │ │ │ └── fm_solvers_unipc.py │ ├── omnigen2 │ │ ├── __init__.py │ │ └── src │ │ │ ├── __init__.py │ │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── attention_processor.py │ │ │ ├── embeddings.py │ │ │ └── transformers │ │ │ │ ├── __init__.py │ │ │ │ ├── block_lumina2.py │ │ │ │ ├── components.py │ │ │ │ ├── repo.py │ │ │ │ └── transformer_omnigen2.py │ │ │ ├── ops │ │ │ └── triton │ │ │ │ ├── __init__.py │ │ │ │ └── layer_norm.py │ │ │ ├── pipelines │ │ │ ├── __init__.py │ │ │ ├── image_processor.py │ │ │ ├── omnigen2 │ │ │ │ ├── pipeline_omnigen2.py │ │ │ │ └── pipeline_omnigen2_chat.py │ │ │ └── pipeline_utils.py │ │ │ ├── schedulers │ │ │ ├── __init__.py │ │ │ ├── scheduling_dpmsolver_multistep.py │ │ │ └── scheduling_flow_match_euler_discrete.py │ │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── img_util.py │ │ │ └── import_utils.py │ ├── qwen_image │ │ ├── __init__.py │ │ ├── qwen_image.py │ │ ├── qwen_image_edit.py │ │ ├── qwen_image_edit_plus.py │ │ └── qwen_image_pipelines.py │ ├── wan22 │ │ ├── __init__.py │ │ ├── wan22_14b_i2v_model.py │ │ ├── wan22_14b_model.py │ │ ├── wan22_5b_model.py │ │ └── wan22_pipeline.py │ └── z_image │ │ ├── __init__.py │ │ └── z_image.py ├── flex2 │ ├── __init__.py │ ├── flex2.py │ └── pipeline.py ├── image_reference_slider_trainer │ ├── ImageReferenceSliderTrainerProcess.py │ ├── __init__.py │ └── config │ │ └── train.example.yaml ├── sd_trainer │ ├── DiffusionTrainer.py │ ├── SDTrainer.py │ ├── UITrainer.py │ ├── __init__.py │ └── config │ │ └── train.example.yaml └── ultimate_slider_trainer │ ├── UltimateSliderTrainerProcess.py │ ├── __init__.py │ └── config │ └── train.example.yaml ├── flux_train_ui.py ├── info.py ├── jobs ├── BaseJob.py ├── ExtensionJob.py ├── ExtractJob.py ├── GenerateJob.py ├── MergeJob.py ├── ModJob.py ├── TrainJob.py ├── __init__.py └── process │ ├── BaseExtensionProcess.py │ ├── BaseExtractProcess.py │ ├── BaseMergeProcess.py │ ├── BaseProcess.py │ ├── BaseSDTrainProcess.py │ ├── BaseTrainProcess.py │ ├── ExtractLoconProcess.py │ ├── ExtractLoraProcess.py │ ├── GenerateProcess.py │ ├── MergeLoconProcess.py │ ├── ModRescaleLoraProcess.py │ ├── TrainESRGANProcess.py │ ├── TrainFineTuneProcess.py │ ├── TrainSDRescaleProcess.py │ ├── TrainSliderProcess.py │ ├── TrainSliderProcessOld.py │ ├── TrainVAEProcess.py │ ├── __init__.py │ └── models │ ├── critic.py │ └── vgg19_critic.py ├── notebooks ├── FLUX_1_dev_LoRA_Training.ipynb ├── FLUX_1_schnell_LoRA_Training.ipynb └── SliderTraining.ipynb ├── output └── .gitkeep ├── requirements.txt ├── run.py ├── run_modal.py ├── scripts ├── calculate_timestep_weighing_flex.py ├── convert_cog.py ├── convert_diffusers_to_comfy.py ├── convert_diffusers_to_comfy_transformer_only.py ├── convert_lora_to_peft_format.py ├── extract_lora_from_flex.py ├── generate_sampler_step_scales.py ├── make_diffusers_model.py ├── make_lcm_sdxl_model.py ├── patch_te_adapter.py ├── repair_dataset_folder.py └── update_sponsors.py ├── test_config.json ├── testing ├── compare_keys.py ├── generate_lora_mapping.py ├── generate_weight_mappings.py ├── merge_in_text_encoder_adapter.py ├── shrink_pixart.py ├── shrink_pixart2.py ├── shrink_pixart_sm.py ├── shrink_pixart_sm2.py ├── shrink_pixart_sm3.py ├── test_bucket_dataloader.py ├── test_model_load_save.py ├── test_vae.py └── test_vae_cycle.py ├── toolkit ├── __init__.py ├── accelerator.py ├── assistant_lora.py ├── basic.py ├── buckets.py ├── civitai.py ├── clip_vision_adapter.py ├── config.py ├── config_modules.py ├── control_generator.py ├── cuda_malloc.py ├── custom_adapter.py ├── data_loader.py ├── data_transfer_object │ └── data_loader.py ├── dataloader_mixins.py ├── dequantize.py ├── ema.py ├── embedding.py ├── esrgan_utils.py ├── extension.py ├── guidance.py ├── image_utils.py ├── inversion_utils.py ├── ip_adapter.py ├── job.py ├── keymaps │ ├── stable_diffusion_refiner.json │ ├── stable_diffusion_refiner_ldm_base.safetensors │ ├── stable_diffusion_refiner_unmatched.json │ ├── stable_diffusion_sd1.json │ ├── stable_diffusion_sd1_ldm_base.safetensors │ ├── stable_diffusion_sd2.json │ ├── stable_diffusion_sd2_ldm_base.safetensors │ ├── stable_diffusion_sd2_unmatched.json │ ├── stable_diffusion_sdxl.json │ ├── stable_diffusion_sdxl_ldm_base.safetensors │ ├── stable_diffusion_sdxl_unmatched.json │ ├── stable_diffusion_ssd.json │ ├── stable_diffusion_ssd_ldm_base.safetensors │ ├── stable_diffusion_ssd_unmatched.json │ ├── stable_diffusion_vega.json │ └── stable_diffusion_vega_ldm_base.safetensors ├── kohya_lora.py ├── kohya_model_util.py ├── layers.py ├── llvae.py ├── logging_aitk.py ├── lora_special.py ├── lorm.py ├── losses.py ├── lycoris_special.py ├── lycoris_utils.py ├── memory_management │ ├── __init__.py │ ├── manager.py │ └── manager_modules.py ├── metadata.py ├── models │ ├── DoRA.py │ ├── FakeVAE.py │ ├── LoRAFormer.py │ ├── RRDB.py │ ├── auraflow.py │ ├── autoencoder_tiny_with_pooled_exits.py │ ├── base_model.py │ ├── block.py │ ├── clip_fusion.py │ ├── clip_pre_processor.py │ ├── cogview4.py │ ├── control_lora_adapter.py │ ├── decorator.py │ ├── diffusion_feature_extraction.py │ ├── flux.py │ ├── flux_sage_attn.py │ ├── i2v_adapter.py │ ├── ilora.py │ ├── ilora2.py │ ├── llm_adapter.py │ ├── loaders │ │ ├── __init__.py │ │ ├── comfy.py │ │ └── umt5.py │ ├── lokr.py │ ├── mean_flow_adapter.py │ ├── pixtral_vision.py │ ├── redux.py │ ├── single_value_adapter.py │ ├── size_agnostic_feature_encoder.py │ ├── sref.py │ ├── subpixel_adapter.py │ ├── te_adapter.py │ ├── te_aug_adapter.py │ ├── vd_adapter.py │ ├── wan21 │ │ ├── __init__.py │ │ ├── autoencoder_kl_wan.py │ │ ├── wan21.py │ │ ├── wan21_i2v.py │ │ ├── wan_attn.py │ │ ├── wan_lora_convert.py │ │ └── wan_utils.py │ └── zipper_resampler.py ├── network_mixins.py ├── optimizer.py ├── optimizers │ ├── adafactor.py │ ├── adam8bit.py │ ├── automagic.py │ ├── optimizer_utils.py │ └── prodigy_8bit.py ├── orig_configs │ └── sd_xl_refiner.yaml ├── paths.py ├── photomaker.py ├── photomaker_pipeline.py ├── pipelines.py ├── pixel_shuffle_encoder.py ├── print.py ├── progress_bar.py ├── prompt_utils.py ├── reference_adapter.py ├── resampler.py ├── sampler.py ├── samplers │ ├── custom_flowmatch_sampler.py │ ├── custom_lcm_scheduler.py │ └── mean_flow_scheduler.py ├── saving.py ├── scheduler.py ├── sd_device_states_presets.py ├── stable_diffusion_model.py ├── style.py ├── timer.py ├── timestep_weighing │ ├── __init__.py │ ├── default_weighing_scheme.py │ └── flex_timestep_weights_plot.png ├── train_pipelines.py ├── train_tools.py ├── unloader.py └── util │ ├── blended_blur_noise.py │ ├── get_model.py │ ├── inverse_cfg.py │ ├── ip_adapter_utils.py │ ├── losses.py │ ├── mask.py │ ├── quantize.py │ ├── shuffle.py │ └── vae.py ├── ui ├── .gitignore ├── README.md ├── cron │ ├── actions │ │ ├── processQueue.ts │ │ └── startJob.ts │ ├── paths.ts │ ├── prisma.ts │ └── worker.ts ├── next.config.ts ├── package-lock.json ├── package.json ├── postcss.config.mjs ├── prisma │ └── schema.prisma ├── public │ ├── doc_workbox_avatar.jpg │ ├── file.svg │ ├── globe.svg │ ├── imgs │ │ ├── diff_guidance.png │ │ ├── diff_guidance_cn.svg │ │ ├── diff_guidance_cn_clean.svg │ │ └── diff_guidance_cn_v2.svg │ ├── next.svg │ ├── ostris_logo.png │ ├── screenshots │ │ ├── adapter-UI0.png │ │ ├── adapter-UI1.png │ │ ├── dashboard_zh.png │ │ └── jobs_new_zh.png │ ├── vercel.svg │ ├── web-app-manifest-192x192.png │ ├── web-app-manifest-512x512.png │ └── window.svg ├── src │ ├── app │ │ ├── api │ │ │ ├── auth │ │ │ │ └── route.ts │ │ │ ├── caption │ │ │ │ └── get │ │ │ │ │ └── route.ts │ │ │ ├── cpu │ │ │ │ └── route.ts │ │ │ ├── datasets │ │ │ │ ├── create │ │ │ │ │ └── route.tsx │ │ │ │ ├── delete │ │ │ │ │ └── route.tsx │ │ │ │ ├── list │ │ │ │ │ └── route.ts │ │ │ │ ├── listImages │ │ │ │ │ └── route.ts │ │ │ │ └── upload │ │ │ │ │ └── route.ts │ │ │ ├── files │ │ │ │ └── [...filePath] │ │ │ │ │ └── route.ts │ │ │ ├── gpu │ │ │ │ └── route.ts │ │ │ ├── img │ │ │ │ ├── [...imagePath] │ │ │ │ │ └── route.ts │ │ │ │ ├── caption │ │ │ │ │ └── route.ts │ │ │ │ ├── delete │ │ │ │ │ └── route.ts │ │ │ │ └── upload │ │ │ │ │ └── route.ts │ │ │ ├── jobs │ │ │ │ ├── [jobID] │ │ │ │ │ ├── delete │ │ │ │ │ │ └── route.ts │ │ │ │ │ ├── files │ │ │ │ │ │ └── route.ts │ │ │ │ │ ├── log │ │ │ │ │ │ └── route.ts │ │ │ │ │ ├── mark_stopped │ │ │ │ │ │ └── route.ts │ │ │ │ │ ├── samples │ │ │ │ │ │ └── route.ts │ │ │ │ │ ├── start │ │ │ │ │ │ └── route.ts │ │ │ │ │ └── stop │ │ │ │ │ │ └── route.ts │ │ │ │ └── route.ts │ │ │ ├── queue │ │ │ │ ├── [queueID] │ │ │ │ │ ├── start │ │ │ │ │ │ └── route.ts │ │ │ │ │ └── stop │ │ │ │ │ │ └── route.ts │ │ │ │ └── route.ts │ │ │ ├── settings │ │ │ │ └── route.ts │ │ │ └── zip │ │ │ │ └── route.ts │ │ ├── apple-icon.png │ │ ├── dashboard │ │ │ └── page.tsx │ │ ├── datasets │ │ │ ├── [datasetName] │ │ │ │ └── page.tsx │ │ │ └── page.tsx │ │ ├── favicon.ico │ │ ├── globals.css │ │ ├── icon.png │ │ ├── icon.svg │ │ ├── jobs │ │ │ ├── [jobID] │ │ │ │ └── page.tsx │ │ │ ├── new │ │ │ │ ├── AdvancedJob.tsx │ │ │ │ ├── SimpleJob.tsx │ │ │ │ ├── jobConfig.ts │ │ │ │ ├── options.ts │ │ │ │ ├── page.tsx │ │ │ │ └── utils.ts │ │ │ └── page.tsx │ │ ├── layout.tsx │ │ ├── manifest.json │ │ ├── page.tsx │ │ └── settings │ │ │ └── page.tsx │ ├── components │ │ ├── AddImagesModal.tsx │ │ ├── AddSingleImageModal.tsx │ │ ├── AuthWrapper.tsx │ │ ├── CPUWidget.tsx │ │ ├── Card.tsx │ │ ├── ConfirmModal.tsx │ │ ├── DatasetImageCard.tsx │ │ ├── DocModal.tsx │ │ ├── ErrorBoundary.tsx │ │ ├── FilesWidget.tsx │ │ ├── FullscreenDropOverlay.tsx │ │ ├── GPUMonitor.tsx │ │ ├── GPUWidget.tsx │ │ ├── JobActionBar.tsx │ │ ├── JobConfigViewer.tsx │ │ ├── JobOverview.tsx │ │ ├── JobsTable.tsx │ │ ├── Loading.tsx │ │ ├── Modal.tsx │ │ ├── SampleControlImage.tsx │ │ ├── SampleImageCard.tsx │ │ ├── SampleImageViewer.tsx │ │ ├── SampleImages.tsx │ │ ├── Sidebar.tsx │ │ ├── SidebarToggleButton.tsx │ │ ├── ThemeProvider.tsx │ │ ├── UniversalTable.tsx │ │ ├── formInputs.tsx │ │ └── layout.tsx │ ├── docs.tsx │ ├── hooks │ │ ├── useCPUInfo.tsx │ │ ├── useDatasetList.tsx │ │ ├── useFilesList.tsx │ │ ├── useFromNull.tsx │ │ ├── useGPUInfo.tsx │ │ ├── useJob.tsx │ │ ├── useJobLog.tsx │ │ ├── useJobsList.tsx │ │ ├── useQueueList.tsx │ │ ├── useSampleImages.tsx │ │ └── useSettings.tsx │ ├── middleware.ts │ ├── paths.ts │ ├── server │ │ └── settings.ts │ ├── types.ts │ └── utils │ │ ├── api.ts │ │ ├── basic.ts │ │ ├── hooks.tsx │ │ ├── jobs.ts │ │ └── queue.ts ├── tailwind.config.ts ├── tsconfig.json └── tsconfig.worker.json └── version.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.vscode/launch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/.vscode/launch.json -------------------------------------------------------------------------------- /DEPLOYMENT_GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/DEPLOYMENT_GUIDE.md -------------------------------------------------------------------------------- /DOCKER_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/DOCKER_README.md -------------------------------------------------------------------------------- /FAQ.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/FAQ.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/README.md -------------------------------------------------------------------------------- /assets/VAE_test1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/assets/VAE_test1.jpg -------------------------------------------------------------------------------- /assets/glif.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/assets/glif.svg -------------------------------------------------------------------------------- /assets/lora_ease_ui.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/assets/lora_ease_ui.png -------------------------------------------------------------------------------- /config/examples/extract.example.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/extract.example.yml -------------------------------------------------------------------------------- /config/examples/generate.example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/generate.example.yaml -------------------------------------------------------------------------------- /config/examples/mod_lora_scale.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/mod_lora_scale.yaml -------------------------------------------------------------------------------- /config/examples/modal/modal_train_lora_flux_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/modal/modal_train_lora_flux_24gb.yaml -------------------------------------------------------------------------------- /config/examples/modal/modal_train_lora_flux_schnell_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/modal/modal_train_lora_flux_schnell_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_flex_redux.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_flex_redux.yaml -------------------------------------------------------------------------------- /config/examples/train_full_fine_tune_flex.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_full_fine_tune_flex.yaml -------------------------------------------------------------------------------- /config/examples/train_full_fine_tune_lumina.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_full_fine_tune_lumina.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_chroma_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_chroma_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_flex2_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_flex2_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_flex_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_flex_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_flux_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_flux_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_flux_kontext_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_flux_kontext_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_flux_schnell_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_flux_schnell_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_hidream_48.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_hidream_48.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_lumina.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_lumina.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_omnigen2_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_omnigen2_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_qwen_image_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_qwen_image_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_qwen_image_edit_2509_32gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_qwen_image_edit_2509_32gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_qwen_image_edit_32gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_qwen_image_edit_32gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_sd35_large_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_sd35_large_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_wan21_14b_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_wan21_14b_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_wan21_1b_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_wan21_1b_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_wan22_14b_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_wan22_14b_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_lora_zimage_24gb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_lora_zimage_24gb.yaml -------------------------------------------------------------------------------- /config/examples/train_slider.example.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/config/examples/train_slider.example.yml -------------------------------------------------------------------------------- /deploy.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/deploy.sh -------------------------------------------------------------------------------- /docker-compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/docker-compose.yml -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/start.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/docker/start.sh -------------------------------------------------------------------------------- /extensions/example/ExampleMergeModels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions/example/ExampleMergeModels.py -------------------------------------------------------------------------------- /extensions/example/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions/example/__init__.py -------------------------------------------------------------------------------- /extensions/example/config/config.example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions/example/config/config.example.yaml -------------------------------------------------------------------------------- /extensions_built_in/advanced_generator/Img2ImgGenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/advanced_generator/Img2ImgGenerator.py -------------------------------------------------------------------------------- /extensions_built_in/advanced_generator/PureLoraGenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/advanced_generator/PureLoraGenerator.py -------------------------------------------------------------------------------- /extensions_built_in/advanced_generator/ReferenceGenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/advanced_generator/ReferenceGenerator.py -------------------------------------------------------------------------------- /extensions_built_in/advanced_generator/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/advanced_generator/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/advanced_generator/config/train.example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/advanced_generator/config/train.example.yaml -------------------------------------------------------------------------------- /extensions_built_in/concept_replacer/ConceptReplacer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/concept_replacer/ConceptReplacer.py -------------------------------------------------------------------------------- /extensions_built_in/concept_replacer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/concept_replacer/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/concept_replacer/config/train.example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/concept_replacer/config/train.example.yaml -------------------------------------------------------------------------------- /extensions_built_in/concept_slider/ConceptSliderTrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/concept_slider/ConceptSliderTrainer.py -------------------------------------------------------------------------------- /extensions_built_in/concept_slider/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/concept_slider/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/dataset_tools/DatasetTools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/dataset_tools/DatasetTools.py -------------------------------------------------------------------------------- /extensions_built_in/dataset_tools/SuperTagger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/dataset_tools/SuperTagger.py -------------------------------------------------------------------------------- /extensions_built_in/dataset_tools/SyncFromCollection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/dataset_tools/SyncFromCollection.py -------------------------------------------------------------------------------- /extensions_built_in/dataset_tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/dataset_tools/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/dataset_tools/tools/caption.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/dataset_tools/tools/caption.py -------------------------------------------------------------------------------- /extensions_built_in/dataset_tools/tools/dataset_tools_config_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/dataset_tools/tools/dataset_tools_config_modules.py -------------------------------------------------------------------------------- /extensions_built_in/dataset_tools/tools/fuyu_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/dataset_tools/tools/fuyu_utils.py -------------------------------------------------------------------------------- /extensions_built_in/dataset_tools/tools/image_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/dataset_tools/tools/image_tools.py -------------------------------------------------------------------------------- /extensions_built_in/dataset_tools/tools/llava_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/dataset_tools/tools/llava_utils.py -------------------------------------------------------------------------------- /extensions_built_in/dataset_tools/tools/sync_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/dataset_tools/tools/sync_tools.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/chroma/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/chroma/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/chroma/chroma_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/chroma/chroma_model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/chroma/chroma_radiance_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/chroma/chroma_radiance_model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/chroma/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/chroma/pipeline.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/chroma/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/chroma/src/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/chroma/src/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/chroma/src/layers.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/chroma/src/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/chroma/src/math.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/chroma/src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/chroma/src/model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/chroma/src/radiance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/chroma/src/radiance.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/f_light/__init__.py: -------------------------------------------------------------------------------- 1 | from .f_light import FLiteModel -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/f_light/f_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/f_light/f_light.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/f_light/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/f_light/src/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/f_light/src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/f_light/src/model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/f_light/src/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/f_light/src/pipeline.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/flux2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/flux2/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/flux2/flux2_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/flux2/flux2_model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/flux2/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/flux2/src/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/flux2/src/autoencoder.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/flux2/src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/flux2/src/model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/flux2/src/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/flux2/src/pipeline.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/flux2/src/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/flux2/src/sampling.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/flux_kontext/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/flux_kontext/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/flux_kontext/flux_kontext.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/hidream_e1_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/hidream_e1_model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/hidream_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/hidream_model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/models/attention.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/models/attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/models/attention_processor.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/models/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/models/embeddings.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/models/moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/models/moe.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/models/transformers/transformer_hidream_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/models/transformers/transformer_hidream_image.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/pipelines/hidream_image/pipeline_hidream_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/pipelines/hidream_image/pipeline_hidream_image.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/pipelines/hidream_image/pipeline_hidream_image_editing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/pipelines/hidream_image/pipeline_hidream_image_editing.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/pipelines/hidream_image/pipeline_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/pipelines/hidream_image/pipeline_output.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/schedulers/flash_flow_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/schedulers/flash_flow_match.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/hidream/src/schedulers/fm_solvers_unipc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/hidream/src/schedulers/fm_solvers_unipc.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/models/attention_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/models/attention_processor.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/models/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/models/embeddings.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/models/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/models/transformers/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/models/transformers/block_lumina2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/models/transformers/block_lumina2.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/models/transformers/components.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/models/transformers/components.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/models/transformers/repo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/models/transformers/repo.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/models/transformers/transformer_omnigen2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/models/transformers/transformer_omnigen2.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/ops/triton/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/ops/triton/layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/ops/triton/layer_norm.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/pipelines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/pipelines/image_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/pipelines/image_processor.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2_chat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/pipelines/omnigen2/pipeline_omnigen2_chat.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/pipelines/pipeline_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/pipelines/pipeline_utils.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/schedulers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/schedulers/scheduling_dpmsolver_multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/schedulers/scheduling_dpmsolver_multistep.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/schedulers/scheduling_flow_match_euler_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/schedulers/scheduling_flow_match_euler_discrete.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/utils/img_util.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/omnigen2/src/utils/import_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/omnigen2/src/utils/import_utils.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/qwen_image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/qwen_image/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/qwen_image/qwen_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/qwen_image/qwen_image.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/qwen_image/qwen_image_edit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/qwen_image/qwen_image_edit.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/qwen_image/qwen_image_edit_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/qwen_image/qwen_image_edit_plus.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/qwen_image/qwen_image_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/qwen_image/qwen_image_pipelines.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/wan22/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/wan22/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/wan22/wan22_14b_i2v_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/wan22/wan22_14b_i2v_model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/wan22/wan22_14b_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/wan22/wan22_14b_model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/wan22/wan22_5b_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/wan22/wan22_5b_model.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/wan22/wan22_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/wan22/wan22_pipeline.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/z_image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/z_image/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/diffusion_models/z_image/z_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/diffusion_models/z_image/z_image.py -------------------------------------------------------------------------------- /extensions_built_in/flex2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/flex2/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/flex2/flex2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/flex2/flex2.py -------------------------------------------------------------------------------- /extensions_built_in/flex2/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/flex2/pipeline.py -------------------------------------------------------------------------------- /extensions_built_in/image_reference_slider_trainer/ImageReferenceSliderTrainerProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/image_reference_slider_trainer/ImageReferenceSliderTrainerProcess.py -------------------------------------------------------------------------------- /extensions_built_in/image_reference_slider_trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/image_reference_slider_trainer/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/image_reference_slider_trainer/config/train.example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/image_reference_slider_trainer/config/train.example.yaml -------------------------------------------------------------------------------- /extensions_built_in/sd_trainer/DiffusionTrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/sd_trainer/DiffusionTrainer.py -------------------------------------------------------------------------------- /extensions_built_in/sd_trainer/SDTrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/sd_trainer/SDTrainer.py -------------------------------------------------------------------------------- /extensions_built_in/sd_trainer/UITrainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/sd_trainer/UITrainer.py -------------------------------------------------------------------------------- /extensions_built_in/sd_trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/sd_trainer/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/sd_trainer/config/train.example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/sd_trainer/config/train.example.yaml -------------------------------------------------------------------------------- /extensions_built_in/ultimate_slider_trainer/UltimateSliderTrainerProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/ultimate_slider_trainer/UltimateSliderTrainerProcess.py -------------------------------------------------------------------------------- /extensions_built_in/ultimate_slider_trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/ultimate_slider_trainer/__init__.py -------------------------------------------------------------------------------- /extensions_built_in/ultimate_slider_trainer/config/train.example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/extensions_built_in/ultimate_slider_trainer/config/train.example.yaml -------------------------------------------------------------------------------- /flux_train_ui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/flux_train_ui.py -------------------------------------------------------------------------------- /info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/info.py -------------------------------------------------------------------------------- /jobs/BaseJob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/BaseJob.py -------------------------------------------------------------------------------- /jobs/ExtensionJob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/ExtensionJob.py -------------------------------------------------------------------------------- /jobs/ExtractJob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/ExtractJob.py -------------------------------------------------------------------------------- /jobs/GenerateJob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/GenerateJob.py -------------------------------------------------------------------------------- /jobs/MergeJob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/MergeJob.py -------------------------------------------------------------------------------- /jobs/ModJob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/ModJob.py -------------------------------------------------------------------------------- /jobs/TrainJob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/TrainJob.py -------------------------------------------------------------------------------- /jobs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/__init__.py -------------------------------------------------------------------------------- /jobs/process/BaseExtensionProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/BaseExtensionProcess.py -------------------------------------------------------------------------------- /jobs/process/BaseExtractProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/BaseExtractProcess.py -------------------------------------------------------------------------------- /jobs/process/BaseMergeProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/BaseMergeProcess.py -------------------------------------------------------------------------------- /jobs/process/BaseProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/BaseProcess.py -------------------------------------------------------------------------------- /jobs/process/BaseSDTrainProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/BaseSDTrainProcess.py -------------------------------------------------------------------------------- /jobs/process/BaseTrainProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/BaseTrainProcess.py -------------------------------------------------------------------------------- /jobs/process/ExtractLoconProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/ExtractLoconProcess.py -------------------------------------------------------------------------------- /jobs/process/ExtractLoraProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/ExtractLoraProcess.py -------------------------------------------------------------------------------- /jobs/process/GenerateProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/GenerateProcess.py -------------------------------------------------------------------------------- /jobs/process/MergeLoconProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/MergeLoconProcess.py -------------------------------------------------------------------------------- /jobs/process/ModRescaleLoraProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/ModRescaleLoraProcess.py -------------------------------------------------------------------------------- /jobs/process/TrainESRGANProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/TrainESRGANProcess.py -------------------------------------------------------------------------------- /jobs/process/TrainFineTuneProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/TrainFineTuneProcess.py -------------------------------------------------------------------------------- /jobs/process/TrainSDRescaleProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/TrainSDRescaleProcess.py -------------------------------------------------------------------------------- /jobs/process/TrainSliderProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/TrainSliderProcess.py -------------------------------------------------------------------------------- /jobs/process/TrainSliderProcessOld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/TrainSliderProcessOld.py -------------------------------------------------------------------------------- /jobs/process/TrainVAEProcess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/TrainVAEProcess.py -------------------------------------------------------------------------------- /jobs/process/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/__init__.py -------------------------------------------------------------------------------- /jobs/process/models/critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/models/critic.py -------------------------------------------------------------------------------- /jobs/process/models/vgg19_critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/jobs/process/models/vgg19_critic.py -------------------------------------------------------------------------------- /notebooks/FLUX_1_dev_LoRA_Training.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/notebooks/FLUX_1_dev_LoRA_Training.ipynb -------------------------------------------------------------------------------- /notebooks/FLUX_1_schnell_LoRA_Training.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/notebooks/FLUX_1_schnell_LoRA_Training.ipynb -------------------------------------------------------------------------------- /notebooks/SliderTraining.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/notebooks/SliderTraining.ipynb -------------------------------------------------------------------------------- /output/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/run.py -------------------------------------------------------------------------------- /run_modal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/run_modal.py -------------------------------------------------------------------------------- /scripts/calculate_timestep_weighing_flex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/calculate_timestep_weighing_flex.py -------------------------------------------------------------------------------- /scripts/convert_cog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/convert_cog.py -------------------------------------------------------------------------------- /scripts/convert_diffusers_to_comfy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/convert_diffusers_to_comfy.py -------------------------------------------------------------------------------- /scripts/convert_diffusers_to_comfy_transformer_only.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/convert_diffusers_to_comfy_transformer_only.py -------------------------------------------------------------------------------- /scripts/convert_lora_to_peft_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/convert_lora_to_peft_format.py -------------------------------------------------------------------------------- /scripts/extract_lora_from_flex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/extract_lora_from_flex.py -------------------------------------------------------------------------------- /scripts/generate_sampler_step_scales.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/generate_sampler_step_scales.py -------------------------------------------------------------------------------- /scripts/make_diffusers_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/make_diffusers_model.py -------------------------------------------------------------------------------- /scripts/make_lcm_sdxl_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/make_lcm_sdxl_model.py -------------------------------------------------------------------------------- /scripts/patch_te_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/patch_te_adapter.py -------------------------------------------------------------------------------- /scripts/repair_dataset_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/repair_dataset_folder.py -------------------------------------------------------------------------------- /scripts/update_sponsors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/scripts/update_sponsors.py -------------------------------------------------------------------------------- /test_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/test_config.json -------------------------------------------------------------------------------- /testing/compare_keys.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/compare_keys.py -------------------------------------------------------------------------------- /testing/generate_lora_mapping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/generate_lora_mapping.py -------------------------------------------------------------------------------- /testing/generate_weight_mappings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/generate_weight_mappings.py -------------------------------------------------------------------------------- /testing/merge_in_text_encoder_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/merge_in_text_encoder_adapter.py -------------------------------------------------------------------------------- /testing/shrink_pixart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/shrink_pixart.py -------------------------------------------------------------------------------- /testing/shrink_pixart2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/shrink_pixart2.py -------------------------------------------------------------------------------- /testing/shrink_pixart_sm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/shrink_pixart_sm.py -------------------------------------------------------------------------------- /testing/shrink_pixart_sm2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/shrink_pixart_sm2.py -------------------------------------------------------------------------------- /testing/shrink_pixart_sm3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/shrink_pixart_sm3.py -------------------------------------------------------------------------------- /testing/test_bucket_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/test_bucket_dataloader.py -------------------------------------------------------------------------------- /testing/test_model_load_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/test_model_load_save.py -------------------------------------------------------------------------------- /testing/test_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/test_vae.py -------------------------------------------------------------------------------- /testing/test_vae_cycle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/testing/test_vae_cycle.py -------------------------------------------------------------------------------- /toolkit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /toolkit/accelerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/accelerator.py -------------------------------------------------------------------------------- /toolkit/assistant_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/assistant_lora.py -------------------------------------------------------------------------------- /toolkit/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/basic.py -------------------------------------------------------------------------------- /toolkit/buckets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/buckets.py -------------------------------------------------------------------------------- /toolkit/civitai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/civitai.py -------------------------------------------------------------------------------- /toolkit/clip_vision_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/clip_vision_adapter.py -------------------------------------------------------------------------------- /toolkit/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/config.py -------------------------------------------------------------------------------- /toolkit/config_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/config_modules.py -------------------------------------------------------------------------------- /toolkit/control_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/control_generator.py -------------------------------------------------------------------------------- /toolkit/cuda_malloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/cuda_malloc.py -------------------------------------------------------------------------------- /toolkit/custom_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/custom_adapter.py -------------------------------------------------------------------------------- /toolkit/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/data_loader.py -------------------------------------------------------------------------------- /toolkit/data_transfer_object/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/data_transfer_object/data_loader.py -------------------------------------------------------------------------------- /toolkit/dataloader_mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/dataloader_mixins.py -------------------------------------------------------------------------------- /toolkit/dequantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/dequantize.py -------------------------------------------------------------------------------- /toolkit/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/ema.py -------------------------------------------------------------------------------- /toolkit/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/embedding.py -------------------------------------------------------------------------------- /toolkit/esrgan_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/esrgan_utils.py -------------------------------------------------------------------------------- /toolkit/extension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/extension.py -------------------------------------------------------------------------------- /toolkit/guidance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/guidance.py -------------------------------------------------------------------------------- /toolkit/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/image_utils.py -------------------------------------------------------------------------------- /toolkit/inversion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/inversion_utils.py -------------------------------------------------------------------------------- /toolkit/ip_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/ip_adapter.py -------------------------------------------------------------------------------- /toolkit/job.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/job.py -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_refiner.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_refiner.json -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_refiner_ldm_base.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_refiner_ldm_base.safetensors -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_refiner_unmatched.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_refiner_unmatched.json -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_sd1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_sd1.json -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_sd1_ldm_base.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_sd1_ldm_base.safetensors -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_sd2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_sd2.json -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_sd2_ldm_base.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_sd2_ldm_base.safetensors -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_sd2_unmatched.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_sd2_unmatched.json -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_sdxl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_sdxl.json -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_sdxl_ldm_base.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_sdxl_ldm_base.safetensors -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_sdxl_unmatched.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_sdxl_unmatched.json -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_ssd.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_ssd.json -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_ssd_ldm_base.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_ssd_ldm_base.safetensors -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_ssd_unmatched.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_ssd_unmatched.json -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_vega.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_vega.json -------------------------------------------------------------------------------- /toolkit/keymaps/stable_diffusion_vega_ldm_base.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/keymaps/stable_diffusion_vega_ldm_base.safetensors -------------------------------------------------------------------------------- /toolkit/kohya_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/kohya_lora.py -------------------------------------------------------------------------------- /toolkit/kohya_model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/kohya_model_util.py -------------------------------------------------------------------------------- /toolkit/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/layers.py -------------------------------------------------------------------------------- /toolkit/llvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/llvae.py -------------------------------------------------------------------------------- /toolkit/logging_aitk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/logging_aitk.py -------------------------------------------------------------------------------- /toolkit/lora_special.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/lora_special.py -------------------------------------------------------------------------------- /toolkit/lorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/lorm.py -------------------------------------------------------------------------------- /toolkit/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/losses.py -------------------------------------------------------------------------------- /toolkit/lycoris_special.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/lycoris_special.py -------------------------------------------------------------------------------- /toolkit/lycoris_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/lycoris_utils.py -------------------------------------------------------------------------------- /toolkit/memory_management/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/memory_management/__init__.py -------------------------------------------------------------------------------- /toolkit/memory_management/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/memory_management/manager.py -------------------------------------------------------------------------------- /toolkit/memory_management/manager_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/memory_management/manager_modules.py -------------------------------------------------------------------------------- /toolkit/metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/metadata.py -------------------------------------------------------------------------------- /toolkit/models/DoRA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/DoRA.py -------------------------------------------------------------------------------- /toolkit/models/FakeVAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/FakeVAE.py -------------------------------------------------------------------------------- /toolkit/models/LoRAFormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/LoRAFormer.py -------------------------------------------------------------------------------- /toolkit/models/RRDB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/RRDB.py -------------------------------------------------------------------------------- /toolkit/models/auraflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/auraflow.py -------------------------------------------------------------------------------- /toolkit/models/autoencoder_tiny_with_pooled_exits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/autoencoder_tiny_with_pooled_exits.py -------------------------------------------------------------------------------- /toolkit/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/base_model.py -------------------------------------------------------------------------------- /toolkit/models/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/block.py -------------------------------------------------------------------------------- /toolkit/models/clip_fusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/clip_fusion.py -------------------------------------------------------------------------------- /toolkit/models/clip_pre_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/clip_pre_processor.py -------------------------------------------------------------------------------- /toolkit/models/cogview4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/cogview4.py -------------------------------------------------------------------------------- /toolkit/models/control_lora_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/control_lora_adapter.py -------------------------------------------------------------------------------- /toolkit/models/decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/decorator.py -------------------------------------------------------------------------------- /toolkit/models/diffusion_feature_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/diffusion_feature_extraction.py -------------------------------------------------------------------------------- /toolkit/models/flux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/flux.py -------------------------------------------------------------------------------- /toolkit/models/flux_sage_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/flux_sage_attn.py -------------------------------------------------------------------------------- /toolkit/models/i2v_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/i2v_adapter.py -------------------------------------------------------------------------------- /toolkit/models/ilora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/ilora.py -------------------------------------------------------------------------------- /toolkit/models/ilora2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/ilora2.py -------------------------------------------------------------------------------- /toolkit/models/llm_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/llm_adapter.py -------------------------------------------------------------------------------- /toolkit/models/loaders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /toolkit/models/loaders/comfy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/loaders/comfy.py -------------------------------------------------------------------------------- /toolkit/models/loaders/umt5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/loaders/umt5.py -------------------------------------------------------------------------------- /toolkit/models/lokr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/lokr.py -------------------------------------------------------------------------------- /toolkit/models/mean_flow_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/mean_flow_adapter.py -------------------------------------------------------------------------------- /toolkit/models/pixtral_vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/pixtral_vision.py -------------------------------------------------------------------------------- /toolkit/models/redux.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/redux.py -------------------------------------------------------------------------------- /toolkit/models/single_value_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/single_value_adapter.py -------------------------------------------------------------------------------- /toolkit/models/size_agnostic_feature_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/size_agnostic_feature_encoder.py -------------------------------------------------------------------------------- /toolkit/models/sref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/sref.py -------------------------------------------------------------------------------- /toolkit/models/subpixel_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/subpixel_adapter.py -------------------------------------------------------------------------------- /toolkit/models/te_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/te_adapter.py -------------------------------------------------------------------------------- /toolkit/models/te_aug_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/te_aug_adapter.py -------------------------------------------------------------------------------- /toolkit/models/vd_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/vd_adapter.py -------------------------------------------------------------------------------- /toolkit/models/wan21/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/wan21/__init__.py -------------------------------------------------------------------------------- /toolkit/models/wan21/autoencoder_kl_wan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/wan21/autoencoder_kl_wan.py -------------------------------------------------------------------------------- /toolkit/models/wan21/wan21.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/wan21/wan21.py -------------------------------------------------------------------------------- /toolkit/models/wan21/wan21_i2v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/wan21/wan21_i2v.py -------------------------------------------------------------------------------- /toolkit/models/wan21/wan_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/wan21/wan_attn.py -------------------------------------------------------------------------------- /toolkit/models/wan21/wan_lora_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/wan21/wan_lora_convert.py -------------------------------------------------------------------------------- /toolkit/models/wan21/wan_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/wan21/wan_utils.py -------------------------------------------------------------------------------- /toolkit/models/zipper_resampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/models/zipper_resampler.py -------------------------------------------------------------------------------- /toolkit/network_mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/network_mixins.py -------------------------------------------------------------------------------- /toolkit/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/optimizer.py -------------------------------------------------------------------------------- /toolkit/optimizers/adafactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/optimizers/adafactor.py -------------------------------------------------------------------------------- /toolkit/optimizers/adam8bit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/optimizers/adam8bit.py -------------------------------------------------------------------------------- /toolkit/optimizers/automagic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/optimizers/automagic.py -------------------------------------------------------------------------------- /toolkit/optimizers/optimizer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/optimizers/optimizer_utils.py -------------------------------------------------------------------------------- /toolkit/optimizers/prodigy_8bit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/optimizers/prodigy_8bit.py -------------------------------------------------------------------------------- /toolkit/orig_configs/sd_xl_refiner.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/orig_configs/sd_xl_refiner.yaml -------------------------------------------------------------------------------- /toolkit/paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/paths.py -------------------------------------------------------------------------------- /toolkit/photomaker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/photomaker.py -------------------------------------------------------------------------------- /toolkit/photomaker_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/photomaker_pipeline.py -------------------------------------------------------------------------------- /toolkit/pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/pipelines.py -------------------------------------------------------------------------------- /toolkit/pixel_shuffle_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/pixel_shuffle_encoder.py -------------------------------------------------------------------------------- /toolkit/print.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/print.py -------------------------------------------------------------------------------- /toolkit/progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/progress_bar.py -------------------------------------------------------------------------------- /toolkit/prompt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/prompt_utils.py -------------------------------------------------------------------------------- /toolkit/reference_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/reference_adapter.py -------------------------------------------------------------------------------- /toolkit/resampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/resampler.py -------------------------------------------------------------------------------- /toolkit/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/sampler.py -------------------------------------------------------------------------------- /toolkit/samplers/custom_flowmatch_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/samplers/custom_flowmatch_sampler.py -------------------------------------------------------------------------------- /toolkit/samplers/custom_lcm_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/samplers/custom_lcm_scheduler.py -------------------------------------------------------------------------------- /toolkit/samplers/mean_flow_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/samplers/mean_flow_scheduler.py -------------------------------------------------------------------------------- /toolkit/saving.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/saving.py -------------------------------------------------------------------------------- /toolkit/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/scheduler.py -------------------------------------------------------------------------------- /toolkit/sd_device_states_presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/sd_device_states_presets.py -------------------------------------------------------------------------------- /toolkit/stable_diffusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/stable_diffusion_model.py -------------------------------------------------------------------------------- /toolkit/style.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/style.py -------------------------------------------------------------------------------- /toolkit/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/timer.py -------------------------------------------------------------------------------- /toolkit/timestep_weighing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /toolkit/timestep_weighing/default_weighing_scheme.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/timestep_weighing/default_weighing_scheme.py -------------------------------------------------------------------------------- /toolkit/timestep_weighing/flex_timestep_weights_plot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/timestep_weighing/flex_timestep_weights_plot.png -------------------------------------------------------------------------------- /toolkit/train_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/train_pipelines.py -------------------------------------------------------------------------------- /toolkit/train_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/train_tools.py -------------------------------------------------------------------------------- /toolkit/unloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/unloader.py -------------------------------------------------------------------------------- /toolkit/util/blended_blur_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/util/blended_blur_noise.py -------------------------------------------------------------------------------- /toolkit/util/get_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/util/get_model.py -------------------------------------------------------------------------------- /toolkit/util/inverse_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/util/inverse_cfg.py -------------------------------------------------------------------------------- /toolkit/util/ip_adapter_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/util/ip_adapter_utils.py -------------------------------------------------------------------------------- /toolkit/util/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/util/losses.py -------------------------------------------------------------------------------- /toolkit/util/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/util/mask.py -------------------------------------------------------------------------------- /toolkit/util/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/util/quantize.py -------------------------------------------------------------------------------- /toolkit/util/shuffle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/util/shuffle.py -------------------------------------------------------------------------------- /toolkit/util/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/toolkit/util/vae.py -------------------------------------------------------------------------------- /ui/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/.gitignore -------------------------------------------------------------------------------- /ui/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/README.md -------------------------------------------------------------------------------- /ui/cron/actions/processQueue.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/cron/actions/processQueue.ts -------------------------------------------------------------------------------- /ui/cron/actions/startJob.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/cron/actions/startJob.ts -------------------------------------------------------------------------------- /ui/cron/paths.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/cron/paths.ts -------------------------------------------------------------------------------- /ui/cron/prisma.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/cron/prisma.ts -------------------------------------------------------------------------------- /ui/cron/worker.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/cron/worker.ts -------------------------------------------------------------------------------- /ui/next.config.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/next.config.ts -------------------------------------------------------------------------------- /ui/package-lock.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/package-lock.json -------------------------------------------------------------------------------- /ui/package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/package.json -------------------------------------------------------------------------------- /ui/postcss.config.mjs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/postcss.config.mjs -------------------------------------------------------------------------------- /ui/prisma/schema.prisma: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/prisma/schema.prisma -------------------------------------------------------------------------------- /ui/public/doc_workbox_avatar.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/doc_workbox_avatar.jpg -------------------------------------------------------------------------------- /ui/public/file.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/file.svg -------------------------------------------------------------------------------- /ui/public/globe.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/globe.svg -------------------------------------------------------------------------------- /ui/public/imgs/diff_guidance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/imgs/diff_guidance.png -------------------------------------------------------------------------------- /ui/public/imgs/diff_guidance_cn.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/imgs/diff_guidance_cn.svg -------------------------------------------------------------------------------- /ui/public/imgs/diff_guidance_cn_clean.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/imgs/diff_guidance_cn_clean.svg -------------------------------------------------------------------------------- /ui/public/imgs/diff_guidance_cn_v2.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/imgs/diff_guidance_cn_v2.svg -------------------------------------------------------------------------------- /ui/public/next.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/next.svg -------------------------------------------------------------------------------- /ui/public/ostris_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/ostris_logo.png -------------------------------------------------------------------------------- /ui/public/screenshots/adapter-UI0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/screenshots/adapter-UI0.png -------------------------------------------------------------------------------- /ui/public/screenshots/adapter-UI1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/screenshots/adapter-UI1.png -------------------------------------------------------------------------------- /ui/public/screenshots/dashboard_zh.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/screenshots/dashboard_zh.png -------------------------------------------------------------------------------- /ui/public/screenshots/jobs_new_zh.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/screenshots/jobs_new_zh.png -------------------------------------------------------------------------------- /ui/public/vercel.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/vercel.svg -------------------------------------------------------------------------------- /ui/public/web-app-manifest-192x192.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/web-app-manifest-192x192.png -------------------------------------------------------------------------------- /ui/public/web-app-manifest-512x512.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/web-app-manifest-512x512.png -------------------------------------------------------------------------------- /ui/public/window.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/public/window.svg -------------------------------------------------------------------------------- /ui/src/app/api/auth/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/auth/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/caption/get/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/caption/get/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/cpu/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/cpu/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/datasets/create/route.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/datasets/create/route.tsx -------------------------------------------------------------------------------- /ui/src/app/api/datasets/delete/route.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/datasets/delete/route.tsx -------------------------------------------------------------------------------- /ui/src/app/api/datasets/list/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/datasets/list/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/datasets/listImages/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/datasets/listImages/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/datasets/upload/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/datasets/upload/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/files/[...filePath]/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/files/[...filePath]/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/gpu/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/gpu/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/img/[...imagePath]/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/img/[...imagePath]/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/img/caption/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/img/caption/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/img/delete/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/img/delete/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/img/upload/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/img/upload/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/jobs/[jobID]/delete/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/jobs/[jobID]/delete/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/jobs/[jobID]/files/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/jobs/[jobID]/files/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/jobs/[jobID]/log/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/jobs/[jobID]/log/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/jobs/[jobID]/mark_stopped/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/jobs/[jobID]/mark_stopped/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/jobs/[jobID]/samples/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/jobs/[jobID]/samples/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/jobs/[jobID]/start/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/jobs/[jobID]/start/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/jobs/[jobID]/stop/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/jobs/[jobID]/stop/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/jobs/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/jobs/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/queue/[queueID]/start/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/queue/[queueID]/start/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/queue/[queueID]/stop/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/queue/[queueID]/stop/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/queue/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/queue/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/settings/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/settings/route.ts -------------------------------------------------------------------------------- /ui/src/app/api/zip/route.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/api/zip/route.ts -------------------------------------------------------------------------------- /ui/src/app/apple-icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/apple-icon.png -------------------------------------------------------------------------------- /ui/src/app/dashboard/page.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/dashboard/page.tsx -------------------------------------------------------------------------------- /ui/src/app/datasets/[datasetName]/page.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/datasets/[datasetName]/page.tsx -------------------------------------------------------------------------------- /ui/src/app/datasets/page.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/datasets/page.tsx -------------------------------------------------------------------------------- /ui/src/app/favicon.ico: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/favicon.ico -------------------------------------------------------------------------------- /ui/src/app/globals.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/globals.css -------------------------------------------------------------------------------- /ui/src/app/icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/icon.png -------------------------------------------------------------------------------- /ui/src/app/icon.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/icon.svg -------------------------------------------------------------------------------- /ui/src/app/jobs/[jobID]/page.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/jobs/[jobID]/page.tsx -------------------------------------------------------------------------------- /ui/src/app/jobs/new/AdvancedJob.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/jobs/new/AdvancedJob.tsx -------------------------------------------------------------------------------- /ui/src/app/jobs/new/SimpleJob.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/jobs/new/SimpleJob.tsx -------------------------------------------------------------------------------- /ui/src/app/jobs/new/jobConfig.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/jobs/new/jobConfig.ts -------------------------------------------------------------------------------- /ui/src/app/jobs/new/options.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/jobs/new/options.ts -------------------------------------------------------------------------------- /ui/src/app/jobs/new/page.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/jobs/new/page.tsx -------------------------------------------------------------------------------- /ui/src/app/jobs/new/utils.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/jobs/new/utils.ts -------------------------------------------------------------------------------- /ui/src/app/jobs/page.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/jobs/page.tsx -------------------------------------------------------------------------------- /ui/src/app/layout.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/layout.tsx -------------------------------------------------------------------------------- /ui/src/app/manifest.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/manifest.json -------------------------------------------------------------------------------- /ui/src/app/page.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/page.tsx -------------------------------------------------------------------------------- /ui/src/app/settings/page.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/app/settings/page.tsx -------------------------------------------------------------------------------- /ui/src/components/AddImagesModal.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/AddImagesModal.tsx -------------------------------------------------------------------------------- /ui/src/components/AddSingleImageModal.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/AddSingleImageModal.tsx -------------------------------------------------------------------------------- /ui/src/components/AuthWrapper.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/AuthWrapper.tsx -------------------------------------------------------------------------------- /ui/src/components/CPUWidget.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/CPUWidget.tsx -------------------------------------------------------------------------------- /ui/src/components/Card.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/Card.tsx -------------------------------------------------------------------------------- /ui/src/components/ConfirmModal.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/ConfirmModal.tsx -------------------------------------------------------------------------------- /ui/src/components/DatasetImageCard.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/DatasetImageCard.tsx -------------------------------------------------------------------------------- /ui/src/components/DocModal.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/DocModal.tsx -------------------------------------------------------------------------------- /ui/src/components/ErrorBoundary.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/ErrorBoundary.tsx -------------------------------------------------------------------------------- /ui/src/components/FilesWidget.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/FilesWidget.tsx -------------------------------------------------------------------------------- /ui/src/components/FullscreenDropOverlay.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/FullscreenDropOverlay.tsx -------------------------------------------------------------------------------- /ui/src/components/GPUMonitor.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/GPUMonitor.tsx -------------------------------------------------------------------------------- /ui/src/components/GPUWidget.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/GPUWidget.tsx -------------------------------------------------------------------------------- /ui/src/components/JobActionBar.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/JobActionBar.tsx -------------------------------------------------------------------------------- /ui/src/components/JobConfigViewer.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/JobConfigViewer.tsx -------------------------------------------------------------------------------- /ui/src/components/JobOverview.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/JobOverview.tsx -------------------------------------------------------------------------------- /ui/src/components/JobsTable.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/JobsTable.tsx -------------------------------------------------------------------------------- /ui/src/components/Loading.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/Loading.tsx -------------------------------------------------------------------------------- /ui/src/components/Modal.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/Modal.tsx -------------------------------------------------------------------------------- /ui/src/components/SampleControlImage.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/SampleControlImage.tsx -------------------------------------------------------------------------------- /ui/src/components/SampleImageCard.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/SampleImageCard.tsx -------------------------------------------------------------------------------- /ui/src/components/SampleImageViewer.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/SampleImageViewer.tsx -------------------------------------------------------------------------------- /ui/src/components/SampleImages.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/SampleImages.tsx -------------------------------------------------------------------------------- /ui/src/components/Sidebar.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/Sidebar.tsx -------------------------------------------------------------------------------- /ui/src/components/SidebarToggleButton.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/SidebarToggleButton.tsx -------------------------------------------------------------------------------- /ui/src/components/ThemeProvider.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/ThemeProvider.tsx -------------------------------------------------------------------------------- /ui/src/components/UniversalTable.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/UniversalTable.tsx -------------------------------------------------------------------------------- /ui/src/components/formInputs.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/formInputs.tsx -------------------------------------------------------------------------------- /ui/src/components/layout.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/components/layout.tsx -------------------------------------------------------------------------------- /ui/src/docs.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/docs.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useCPUInfo.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useCPUInfo.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useDatasetList.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useDatasetList.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useFilesList.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useFilesList.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useFromNull.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useFromNull.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useGPUInfo.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useGPUInfo.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useJob.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useJob.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useJobLog.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useJobLog.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useJobsList.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useJobsList.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useQueueList.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useQueueList.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useSampleImages.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useSampleImages.tsx -------------------------------------------------------------------------------- /ui/src/hooks/useSettings.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/hooks/useSettings.tsx -------------------------------------------------------------------------------- /ui/src/middleware.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/middleware.ts -------------------------------------------------------------------------------- /ui/src/paths.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/paths.ts -------------------------------------------------------------------------------- /ui/src/server/settings.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/server/settings.ts -------------------------------------------------------------------------------- /ui/src/types.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/types.ts -------------------------------------------------------------------------------- /ui/src/utils/api.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/utils/api.ts -------------------------------------------------------------------------------- /ui/src/utils/basic.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/utils/basic.ts -------------------------------------------------------------------------------- /ui/src/utils/hooks.tsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/utils/hooks.tsx -------------------------------------------------------------------------------- /ui/src/utils/jobs.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/utils/jobs.ts -------------------------------------------------------------------------------- /ui/src/utils/queue.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/src/utils/queue.ts -------------------------------------------------------------------------------- /ui/tailwind.config.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/tailwind.config.ts -------------------------------------------------------------------------------- /ui/tsconfig.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/tsconfig.json -------------------------------------------------------------------------------- /ui/tsconfig.worker.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocWorkBox/ai-toolkit-easy2use/HEAD/ui/tsconfig.worker.json -------------------------------------------------------------------------------- /version.py: -------------------------------------------------------------------------------- 1 | VERSION = "0.7.7" --------------------------------------------------------------------------------