├── .github └── FUNDING.yml ├── .gitignore ├── LICENSE.Apache-2.0 ├── LICENSE.GAPLV3 ├── LICENSE.MIT ├── README.md ├── StableTuner.cmd ├── configs ├── 4090_SD2_512_dreambooth.json ├── 4090_SD2_512_finetune.json └── test.json ├── environment.yaml ├── install_stabletuner.bat ├── requirements.txt ├── resources ├── accelerate_windows │ └── accelerate_default_config.yaml ├── bitsandbytes_windows │ ├── cextension.py │ ├── libbitsandbytes_cpu.dll │ ├── libbitsandbytes_cuda116.dll │ └── main.py ├── stableTuner_icon.png ├── stableTuner_logo.png └── stableTuner_notebook.ipynb └── scripts ├── __init__.py ├── captionBuddy.py ├── clip_segmentation.py ├── configuration_gui.py ├── convert_diffusers_to_sd_cli.py ├── converters.py ├── dataloaders_util.py ├── lion_pytorch ├── __init__.py ├── lion_pytorch.py └── triton.py ├── lora_utils.py ├── model_util.py ├── trainer.py ├── trainer_util.py └── windows_install.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.Apache-2.0: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/LICENSE.Apache-2.0 -------------------------------------------------------------------------------- /LICENSE.GAPLV3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/LICENSE.GAPLV3 -------------------------------------------------------------------------------- /LICENSE.MIT: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/LICENSE.MIT -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/README.md -------------------------------------------------------------------------------- /StableTuner.cmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/StableTuner.cmd -------------------------------------------------------------------------------- /configs/4090_SD2_512_dreambooth.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/configs/4090_SD2_512_dreambooth.json -------------------------------------------------------------------------------- /configs/4090_SD2_512_finetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/configs/4090_SD2_512_finetune.json -------------------------------------------------------------------------------- /configs/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/configs/test.json -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/environment.yaml -------------------------------------------------------------------------------- /install_stabletuner.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/install_stabletuner.bat -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/requirements.txt -------------------------------------------------------------------------------- /resources/accelerate_windows/accelerate_default_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/resources/accelerate_windows/accelerate_default_config.yaml -------------------------------------------------------------------------------- /resources/bitsandbytes_windows/cextension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/resources/bitsandbytes_windows/cextension.py -------------------------------------------------------------------------------- /resources/bitsandbytes_windows/libbitsandbytes_cpu.dll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/resources/bitsandbytes_windows/libbitsandbytes_cpu.dll -------------------------------------------------------------------------------- /resources/bitsandbytes_windows/libbitsandbytes_cuda116.dll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/resources/bitsandbytes_windows/libbitsandbytes_cuda116.dll -------------------------------------------------------------------------------- /resources/bitsandbytes_windows/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/resources/bitsandbytes_windows/main.py -------------------------------------------------------------------------------- /resources/stableTuner_icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/resources/stableTuner_icon.png -------------------------------------------------------------------------------- /resources/stableTuner_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/resources/stableTuner_logo.png -------------------------------------------------------------------------------- /resources/stableTuner_notebook.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/resources/stableTuner_notebook.ipynb -------------------------------------------------------------------------------- /scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/captionBuddy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/captionBuddy.py -------------------------------------------------------------------------------- /scripts/clip_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/clip_segmentation.py -------------------------------------------------------------------------------- /scripts/configuration_gui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/configuration_gui.py -------------------------------------------------------------------------------- /scripts/convert_diffusers_to_sd_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/convert_diffusers_to_sd_cli.py -------------------------------------------------------------------------------- /scripts/converters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/converters.py -------------------------------------------------------------------------------- /scripts/dataloaders_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/dataloaders_util.py -------------------------------------------------------------------------------- /scripts/lion_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/lion_pytorch/__init__.py -------------------------------------------------------------------------------- /scripts/lion_pytorch/lion_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/lion_pytorch/lion_pytorch.py -------------------------------------------------------------------------------- /scripts/lion_pytorch/triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/lion_pytorch/triton.py -------------------------------------------------------------------------------- /scripts/lora_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/lora_utils.py -------------------------------------------------------------------------------- /scripts/model_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/model_util.py -------------------------------------------------------------------------------- /scripts/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/trainer.py -------------------------------------------------------------------------------- /scripts/trainer_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/trainer_util.py -------------------------------------------------------------------------------- /scripts/windows_install.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/devilismyfriend/StableTuner/HEAD/scripts/windows_install.py --------------------------------------------------------------------------------