From 2ec7432440c3ce1f949572016cebe7e5fbeae705 Mon Sep 17 00:00:00 2001 From: bmaltais Date: Sun, 29 Jan 2023 11:17:30 -0500 Subject: [PATCH] Fix issue 81: https://github.com/bmaltais/kohya_ss/issues/81 --- README.md | 4 ++-- tools/cudann_1.8_install.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 764c375..4342fe9 100644 --- a/README.md +++ b/README.md @@ -2,7 +2,7 @@ This repository repository is providing a Windows focussed Gradio GUI for kohya's Stable Diffusion trainers found here: https://github.com/kohya-ss/sd-scripts. The GUI allow you to set the training parameters and generate and run the required CLI command to train the model. -If you run on linux and would like to use the GUI there is now a port of it as a docker container. You can find the project here: https://github.com/P2Enjoy/kohya_ss-docker +If you run on Linux and would like to use the GUI there is now a port of it as a docker container. You can find the project here: https://github.com/P2Enjoy/kohya_ss-docker ## Tutorials @@ -57,7 +57,7 @@ This step is optional but can improve the learning speed for NVidia 30X0/40X0 ow Due to the filesize I can't host the DLLs needed for CUDNN 8.6 on Github, I strongly advise you download them for a speed boost in sample generation (almost 50% on 4090) you can download them from here: https://b1.thefileditch.ch/mwxKTEtelILoIbMbruuM.zip -To install simply unzip the directory and place the cudnn_windows folder in the root of the kohya_diffusers_fine_tuning repo. +To install simply unzip the directory and place the `cudnn_windows` folder in the root of the kohya_ss repo. Run the following command to install: diff --git a/tools/cudann_1.8_install.py b/tools/cudann_1.8_install.py index 75aef92..dec38a1 100644 --- a/tools/cudann_1.8_install.py +++ b/tools/cudann_1.8_install.py @@ -100,5 +100,7 @@ if os.name == "nt": if os.path.exists(dest_file): shutil.copy2(src_file, cudnn_dest) print("Copied CUDNN 8.6 files to destination") + else: + print(f"Installation Failed: \"{cudnn_src}\" could not be found. ") \ No newline at end of file