Skip to content

Commit

Permalink
feat: added anylora and animepasteldream as pretrained model
Browse files Browse the repository at this point in the history
  • Loading branch information
@Linaqruf
Linaqruf committed Mar 28, 2023
1 parent 2f14cc5 commit 061c953
Show file tree
Hide file tree
Showing 4 changed files with 48 additions and 40 deletions.
22 changes: 12 additions & 10 deletions kohya-LoRA-dreambooth.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -209,7 209,7 @@
" os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"3\"\n",
" os.environ[\"BITSANDBYTES_NOWELCOME\"] = \"1\" \n",
" os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:$LD_LIBRARY_PATH\"\n",
"\n",
" os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
"\n",
"main()\n"
]
Expand Down Expand Up @@ -275,23 275,23 @@
" \"\",\n",
" \"https://huggingface.co/Linaqruf/personal-backup/resolve/main/models/animefull-final-pruned.ckpt\",\n",
" \"https://huggingface.co/cag/anything-v3-1/resolve/main/anything-v3-1.safetensors\",\n",
" \"https://huggingface.co/cag/anything-v3-3/resolve/main/anything-v3-3.safetensors\",\n",
" \"https://huggingface.co/Lykon/AnyLoRA/resolve/main/AnyLoRA_noVae_fp16.safetensors\",\n",
" \"https://huggingface.co/Lykon/AnimePastelDream/resolve/main/AnimePastelDream_Soft_noVae_fp16.safetensors\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/chillout_mix-pruned.safetensors\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/Counterfeit-V2-5-pruned.safetensors\",\n",
" \"https://huggingface.co/prompthero/openjourney-v2/resolve/main/openjourney-v2.ckpt\",\n",
" \"https://huggingface.co/prompthero/openjourney-v4/resolve/main/openjourney-v4.ckpt\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/stable_diffusion_1_5-pruned.safetensors\",\n",
"]\n",
"modelList = [\n",
" \"\",\n",
" \"Animefull-final-pruned\",\n",
" \"Anything-v3-1\",\n",
" \"Anything-v3-3\",\n",
" \"AnyLoRA\",\n",
" \"AnimePastelDream\", \n",
" \"Chillout-mix\",\n",
" \"Counterfeit-v2-5\",\n",
" \"OpenJourney-v2\",\n",
" \"OpenJourney-v4\",\n",
" \"Stable-Diffusion-v1-5\",\n",
"]\n",
"modelName = \"Anything-v3-1\" # @param [\"\", \"Animefull-final-pruned\", \"Anything-v3-1\", \"Anything-v3-3\", \"Chillout-mix\", \"Counterfeit-v2-5\", \"OpenJourney-v2\", \"Stable-Diffusion-v1-5\"]\n",
"modelName = \"Anything-v3-1\" # @param [\"\", \"Animefull-final-pruned\", \"Anything-v3-1\", \"AnyLoRA\", \"AnimePastelDream\", \"Chillout-mix\", \"OpenJourney-v4\", \"Stable-Diffusion-v1-5\"]\n",
"\n",
"# @markdown ### SD2.x model\n",
"v2ModelUrl = [\n",
Expand All @@ -301,6 301,7 @@
" \"https://huggingface.co/p1atdev/pd-archive/resolve/main/plat-v1-3-1.safetensors\",\n",
" \"https://huggingface.co/gsdf/Replicant-V1.0/resolve/main/Replicant-V1.0.safetensors\",\n",
" \"https://huggingface.co/IlluminatiAI/Illuminati_Diffusion_v1.0/resolve/main/illuminati_diffusion_v1.0.safetensors\",\n",
" \"https://huggingface.co/4eJIoBek/Illuminati-Diffusion-v1-1/resolve/main/illuminatiDiffusionV1_v11.safetensors\",\n",
" \"https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/wd-1-4-anime_e2.ckpt\",\n",
" \"https://huggingface.co/waifu-diffusion/wd-1-5-beta2/resolve/main/checkpoints/wd-1-5-beta2-fp32.safetensors\",\n",
" \"https://huggingface.co/waifu-diffusion/wd-1-5-beta2/resolve/main/checkpoints/wd-1-5-beta2-aesthetic-fp32.safetensors\",\n",
Expand All @@ -311,12 312,13 @@
" \"stable-diffusion-2-1-768v\",\n",
" \"plat-diffusion-v1-3-1\",\n",
" \"replicant-v1\",\n",
" \"illuminati-diffusion-v1-768v\",\n",
" \"illuminati-diffusion-v1-0\",\n",
" \"illuminati-diffusion-v1-1\",\n",
" \"waifu-diffusion-1-4-anime-e2\",\n",
" \"waifu-diffusion-1-5-e2\",\n",
" \"waifu-diffusion-1-5-e2-aesthetic\",\n",
"]\n",
"v2ModelName = \"\" # @param [\"\", \"stable-diffusion-2-1-base\", \"stable-diffusion-2-1-768v\", \"plat-diffusion-v1-3-1\", \"replicant-v1\", \"illuminati-diffusion-v1-768v\", \"waifu-diffusion-1-4-anime-e2\", \"waifu-diffusion-1-5-e2\", \"waifu-diffusion-1-5-e2-aesthetic\"]\n",
"v2ModelName = \"\" # @param [\"\", \"stable-diffusion-2-1-base\", \"stable-diffusion-2-1-768v\", \"plat-diffusion-v1-3-1\", \"replicant-v1\", \"illuminati-diffusion-v1-0\", \"illuminati-diffusion-v1-1\", \"waifu-diffusion-1-4-anime-e2\", \"waifu-diffusion-1-5-e2\", \"waifu-diffusion-1-5-e2-aesthetic\"]\n",
"\n",
"if modelName:\n",
" installModels.append((modelName, modelUrl[modelList.index(modelName)]))\n",
Expand Down
22 changes: 12 additions & 10 deletions kohya-LoRA-finetuner.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -209,7 209,7 @@
" os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"3\"\n",
" os.environ[\"BITSANDBYTES_NOWELCOME\"] = \"1\" \n",
" os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:$LD_LIBRARY_PATH\"\n",
"\n",
" os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
"\n",
"main()\n"
]
Expand Down Expand Up @@ -275,23 275,23 @@
" \"\",\n",
" \"https://huggingface.co/Linaqruf/personal-backup/resolve/main/models/animefull-final-pruned.ckpt\",\n",
" \"https://huggingface.co/cag/anything-v3-1/resolve/main/anything-v3-1.safetensors\",\n",
" \"https://huggingface.co/cag/anything-v3-3/resolve/main/anything-v3-3.safetensors\",\n",
" \"https://huggingface.co/Lykon/AnyLoRA/resolve/main/AnyLoRA_noVae_fp16.safetensors\",\n",
" \"https://huggingface.co/Lykon/AnimePastelDream/resolve/main/AnimePastelDream_Soft_noVae_fp16.safetensors\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/chillout_mix-pruned.safetensors\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/Counterfeit-V2-5-pruned.safetensors\",\n",
" \"https://huggingface.co/prompthero/openjourney-v2/resolve/main/openjourney-v2.ckpt\",\n",
" \"https://huggingface.co/prompthero/openjourney-v4/resolve/main/openjourney-v4.ckpt\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/stable_diffusion_1_5-pruned.safetensors\",\n",
"]\n",
"modelList = [\n",
" \"\",\n",
" \"Animefull-final-pruned\",\n",
" \"Anything-v3-1\",\n",
" \"Anything-v3-3\",\n",
" \"AnyLoRA\",\n",
" \"AnimePastelDream\", \n",
" \"Chillout-mix\",\n",
" \"Counterfeit-v2-5\",\n",
" \"OpenJourney-v2\",\n",
" \"OpenJourney-v4\",\n",
" \"Stable-Diffusion-v1-5\",\n",
"]\n",
"modelName = \"Anything-v3-1\" # @param [\"\", \"Animefull-final-pruned\", \"Anything-v3-1\", \"Anything-v3-3\", \"Chillout-mix\", \"Counterfeit-v2-5\", \"OpenJourney-v2\", \"Stable-Diffusion-v1-5\"]\n",
"modelName = \"Anything-v3-1\" # @param [\"\", \"Animefull-final-pruned\", \"Anything-v3-1\", \"AnyLoRA\", \"AnimePastelDream\", \"Chillout-mix\", \"OpenJourney-v4\", \"Stable-Diffusion-v1-5\"]\n",
"\n",
"# @markdown ### SD2.x model\n",
"v2ModelUrl = [\n",
Expand All @@ -301,6 301,7 @@
" \"https://huggingface.co/p1atdev/pd-archive/resolve/main/plat-v1-3-1.safetensors\",\n",
" \"https://huggingface.co/gsdf/Replicant-V1.0/resolve/main/Replicant-V1.0.safetensors\",\n",
" \"https://huggingface.co/IlluminatiAI/Illuminati_Diffusion_v1.0/resolve/main/illuminati_diffusion_v1.0.safetensors\",\n",
" \"https://huggingface.co/4eJIoBek/Illuminati-Diffusion-v1-1/resolve/main/illuminatiDiffusionV1_v11.safetensors\",\n",
" \"https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/wd-1-4-anime_e2.ckpt\",\n",
" \"https://huggingface.co/waifu-diffusion/wd-1-5-beta2/resolve/main/checkpoints/wd-1-5-beta2-fp32.safetensors\",\n",
" \"https://huggingface.co/waifu-diffusion/wd-1-5-beta2/resolve/main/checkpoints/wd-1-5-beta2-aesthetic-fp32.safetensors\",\n",
Expand All @@ -311,12 312,13 @@
" \"stable-diffusion-2-1-768v\",\n",
" \"plat-diffusion-v1-3-1\",\n",
" \"replicant-v1\",\n",
" \"illuminati-diffusion-v1-768v\",\n",
" \"illuminati-diffusion-v1-0\",\n",
" \"illuminati-diffusion-v1-1\",\n",
" \"waifu-diffusion-1-4-anime-e2\",\n",
" \"waifu-diffusion-1-5-e2\",\n",
" \"waifu-diffusion-1-5-e2-aesthetic\",\n",
"]\n",
"v2ModelName = \"\" # @param [\"\", \"stable-diffusion-2-1-base\", \"stable-diffusion-2-1-768v\", \"plat-diffusion-v1-3-1\", \"replicant-v1\", \"illuminati-diffusion-v1-768v\", \"waifu-diffusion-1-4-anime-e2\", \"waifu-diffusion-1-5-e2\", \"waifu-diffusion-1-5-e2-aesthetic\"]\n",
"v2ModelName = \"\" # @param [\"\", \"stable-diffusion-2-1-base\", \"stable-diffusion-2-1-768v\", \"plat-diffusion-v1-3-1\", \"replicant-v1\", \"illuminati-diffusion-v1-0\", \"illuminati-diffusion-v1-1\", \"waifu-diffusion-1-4-anime-e2\", \"waifu-diffusion-1-5-e2\", \"waifu-diffusion-1-5-e2-aesthetic\"]\n",
"\n",
"if modelName:\n",
" installModels.append((modelName, modelUrl[modelList.index(modelName)]))\n",
Expand Down
22 changes: 12 additions & 10 deletions kohya-dreambooth.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -209,7 209,7 @@
" os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"3\"\n",
" os.environ[\"BITSANDBYTES_NOWELCOME\"] = \"1\" \n",
" os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:$LD_LIBRARY_PATH\"\n",
"\n",
" os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
"\n",
"main()\n"
]
Expand Down Expand Up @@ -275,23 275,23 @@
" \"\",\n",
" \"https://huggingface.co/Linaqruf/personal-backup/resolve/main/models/animefull-final-pruned.ckpt\",\n",
" \"https://huggingface.co/cag/anything-v3-1/resolve/main/anything-v3-1.safetensors\",\n",
" \"https://huggingface.co/cag/anything-v3-3/resolve/main/anything-v3-3.safetensors\",\n",
" \"https://huggingface.co/Lykon/AnyLoRA/resolve/main/AnyLoRA_noVae_fp16.safetensors\",\n",
" \"https://huggingface.co/Lykon/AnimePastelDream/resolve/main/AnimePastelDream_Soft_noVae_fp16.safetensors\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/chillout_mix-pruned.safetensors\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/Counterfeit-V2-5-pruned.safetensors\",\n",
" \"https://huggingface.co/prompthero/openjourney-v2/resolve/main/openjourney-v2.ckpt\",\n",
" \"https://huggingface.co/prompthero/openjourney-v4/resolve/main/openjourney-v4.ckpt\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/stable_diffusion_1_5-pruned.safetensors\",\n",
"]\n",
"modelList = [\n",
" \"\",\n",
" \"Animefull-final-pruned\",\n",
" \"Anything-v3-1\",\n",
" \"Anything-v3-3\",\n",
" \"AnyLoRA\",\n",
" \"AnimePastelDream\", \n",
" \"Chillout-mix\",\n",
" \"Counterfeit-v2-5\",\n",
" \"OpenJourney-v2\",\n",
" \"OpenJourney-v4\",\n",
" \"Stable-Diffusion-v1-5\",\n",
"]\n",
"modelName = \"Anything-v3-1\" # @param [\"\", \"Animefull-final-pruned\", \"Anything-v3-1\", \"Anything-v3-3\", \"Chillout-mix\", \"Counterfeit-v2-5\", \"OpenJourney-v2\", \"Stable-Diffusion-v1-5\"]\n",
"modelName = \"Anything-v3-1\" # @param [\"\", \"Animefull-final-pruned\", \"Anything-v3-1\", \"AnyLoRA\", \"AnimePastelDream\", \"Chillout-mix\", \"OpenJourney-v4\", \"Stable-Diffusion-v1-5\"]\n",
"\n",
"# @markdown ### SD2.x model\n",
"v2ModelUrl = [\n",
Expand All @@ -301,6 301,7 @@
" \"https://huggingface.co/p1atdev/pd-archive/resolve/main/plat-v1-3-1.safetensors\",\n",
" \"https://huggingface.co/gsdf/Replicant-V1.0/resolve/main/Replicant-V1.0.safetensors\",\n",
" \"https://huggingface.co/IlluminatiAI/Illuminati_Diffusion_v1.0/resolve/main/illuminati_diffusion_v1.0.safetensors\",\n",
" \"https://huggingface.co/4eJIoBek/Illuminati-Diffusion-v1-1/resolve/main/illuminatiDiffusionV1_v11.safetensors\",\n",
" \"https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/wd-1-4-anime_e2.ckpt\",\n",
" \"https://huggingface.co/waifu-diffusion/wd-1-5-beta2/resolve/main/checkpoints/wd-1-5-beta2-fp32.safetensors\",\n",
" \"https://huggingface.co/waifu-diffusion/wd-1-5-beta2/resolve/main/checkpoints/wd-1-5-beta2-aesthetic-fp32.safetensors\",\n",
Expand All @@ -311,12 312,13 @@
" \"stable-diffusion-2-1-768v\",\n",
" \"plat-diffusion-v1-3-1\",\n",
" \"replicant-v1\",\n",
" \"illuminati-diffusion-v1-768v\",\n",
" \"illuminati-diffusion-v1-0\",\n",
" \"illuminati-diffusion-v1-1\",\n",
" \"waifu-diffusion-1-4-anime-e2\",\n",
" \"waifu-diffusion-1-5-e2\",\n",
" \"waifu-diffusion-1-5-e2-aesthetic\",\n",
"]\n",
"v2ModelName = \"\" # @param [\"\", \"stable-diffusion-2-1-base\", \"stable-diffusion-2-1-768v\", \"plat-diffusion-v1-3-1\", \"replicant-v1\", \"illuminati-diffusion-v1-768v\", \"waifu-diffusion-1-4-anime-e2\", \"waifu-diffusion-1-5-e2\", \"waifu-diffusion-1-5-e2-aesthetic\"]\n",
"v2ModelName = \"\" # @param [\"\", \"stable-diffusion-2-1-base\", \"stable-diffusion-2-1-768v\", \"plat-diffusion-v1-3-1\", \"replicant-v1\", \"illuminati-diffusion-v1-0\", \"illuminati-diffusion-v1-1\", \"waifu-diffusion-1-4-anime-e2\", \"waifu-diffusion-1-5-e2\", \"waifu-diffusion-1-5-e2-aesthetic\"]\n",
"\n",
"if modelName:\n",
" installModels.append((modelName, modelUrl[modelList.index(modelName)]))\n",
Expand Down
22 changes: 12 additions & 10 deletions kohya-trainer.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -209,7 209,7 @@
" os.environ[\"TF_CPP_MIN_LOG_LEVEL\"] = \"3\"\n",
" os.environ[\"BITSANDBYTES_NOWELCOME\"] = \"1\" \n",
" os.environ[\"LD_LIBRARY_PATH\"] = \"/usr/local/cuda/lib64/:$LD_LIBRARY_PATH\"\n",
"\n",
" os.environ[\"SAFETENSORS_FAST_GPU\"] = \"1\"\n",
"\n",
"main()\n"
]
Expand Down Expand Up @@ -275,23 275,23 @@
" \"\",\n",
" \"https://huggingface.co/Linaqruf/personal-backup/resolve/main/models/animefull-final-pruned.ckpt\",\n",
" \"https://huggingface.co/cag/anything-v3-1/resolve/main/anything-v3-1.safetensors\",\n",
" \"https://huggingface.co/cag/anything-v3-3/resolve/main/anything-v3-3.safetensors\",\n",
" \"https://huggingface.co/Lykon/AnyLoRA/resolve/main/AnyLoRA_noVae_fp16.safetensors\",\n",
" \"https://huggingface.co/Lykon/AnimePastelDream/resolve/main/AnimePastelDream_Soft_noVae_fp16.safetensors\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/chillout_mix-pruned.safetensors\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/Counterfeit-V2-5-pruned.safetensors\",\n",
" \"https://huggingface.co/prompthero/openjourney-v2/resolve/main/openjourney-v2.ckpt\",\n",
" \"https://huggingface.co/prompthero/openjourney-v4/resolve/main/openjourney-v4.ckpt\",\n",
" \"https://huggingface.co/Linaqruf/stolen/resolve/main/pruned-models/stable_diffusion_1_5-pruned.safetensors\",\n",
"]\n",
"modelList = [\n",
" \"\",\n",
" \"Animefull-final-pruned\",\n",
" \"Anything-v3-1\",\n",
" \"Anything-v3-3\",\n",
" \"AnyLoRA\",\n",
" \"AnimePastelDream\", \n",
" \"Chillout-mix\",\n",
" \"Counterfeit-v2-5\",\n",
" \"OpenJourney-v2\",\n",
" \"OpenJourney-v4\",\n",
" \"Stable-Diffusion-v1-5\",\n",
"]\n",
"modelName = \"Anything-v3-1\" # @param [\"\", \"Animefull-final-pruned\", \"Anything-v3-1\", \"Anything-v3-3\", \"Chillout-mix\", \"Counterfeit-v2-5\", \"OpenJourney-v2\", \"Stable-Diffusion-v1-5\"]\n",
"modelName = \"Anything-v3-1\" # @param [\"\", \"Animefull-final-pruned\", \"Anything-v3-1\", \"AnyLoRA\", \"AnimePastelDream\", \"Chillout-mix\", \"OpenJourney-v4\", \"Stable-Diffusion-v1-5\"]\n",
"\n",
"# @markdown ### SD2.x model\n",
"v2ModelUrl = [\n",
Expand All @@ -301,6 301,7 @@
" \"https://huggingface.co/p1atdev/pd-archive/resolve/main/plat-v1-3-1.safetensors\",\n",
" \"https://huggingface.co/gsdf/Replicant-V1.0/resolve/main/Replicant-V1.0.safetensors\",\n",
" \"https://huggingface.co/IlluminatiAI/Illuminati_Diffusion_v1.0/resolve/main/illuminati_diffusion_v1.0.safetensors\",\n",
" \"https://huggingface.co/4eJIoBek/Illuminati-Diffusion-v1-1/resolve/main/illuminatiDiffusionV1_v11.safetensors\",\n",
" \"https://huggingface.co/hakurei/waifu-diffusion-v1-4/resolve/main/wd-1-4-anime_e2.ckpt\",\n",
" \"https://huggingface.co/waifu-diffusion/wd-1-5-beta2/resolve/main/checkpoints/wd-1-5-beta2-fp32.safetensors\",\n",
" \"https://huggingface.co/waifu-diffusion/wd-1-5-beta2/resolve/main/checkpoints/wd-1-5-beta2-aesthetic-fp32.safetensors\",\n",
Expand All @@ -311,12 312,13 @@
" \"stable-diffusion-2-1-768v\",\n",
" \"plat-diffusion-v1-3-1\",\n",
" \"replicant-v1\",\n",
" \"illuminati-diffusion-v1-768v\",\n",
" \"illuminati-diffusion-v1-0\",\n",
" \"illuminati-diffusion-v1-1\",\n",
" \"waifu-diffusion-1-4-anime-e2\",\n",
" \"waifu-diffusion-1-5-e2\",\n",
" \"waifu-diffusion-1-5-e2-aesthetic\",\n",
"]\n",
"v2ModelName = \"\" # @param [\"\", \"stable-diffusion-2-1-base\", \"stable-diffusion-2-1-768v\", \"plat-diffusion-v1-3-1\", \"replicant-v1\", \"illuminati-diffusion-v1-768v\", \"waifu-diffusion-1-4-anime-e2\", \"waifu-diffusion-1-5-e2\", \"waifu-diffusion-1-5-e2-aesthetic\"]\n",
"v2ModelName = \"\" # @param [\"\", \"stable-diffusion-2-1-base\", \"stable-diffusion-2-1-768v\", \"plat-diffusion-v1-3-1\", \"replicant-v1\", \"illuminati-diffusion-v1-0\", \"illuminati-diffusion-v1-1\", \"waifu-diffusion-1-4-anime-e2\", \"waifu-diffusion-1-5-e2\", \"waifu-diffusion-1-5-e2-aesthetic\"]\n",
"\n",
"if modelName:\n",
" installModels.append((modelName, modelUrl[modelList.index(modelName)]))\n",
Expand Down

0 comments on commit 061c953

Please sign in to comment.