diff --git a/automatic.sh b/automatic.sh index 616260659..9d769fbfe 100755 --- a/automatic.sh +++ b/automatic.sh @@ -8,15 +8,13 @@ export CUDA_LAUNCH_BLOCKING=0 export CUDA_CACHE_DISABLE=0 export CUDA_AUTO_BOOST=1 export CUDA_DEVICE_DEFAULT_PERSISTING_L2_CACHE_PERCENTAGE_LIMIT=0 -# export LD_PRELOAD=libtcmalloc.so -# TORCH_CUDA_ARCH_LIST="8.6" if [ "$PYTHON" == "" ]; then PYTHON=$(which python) fi -CMD="launch.py --api --xformers --disable-console-progressbars --gradio-queue --skip-version-check --cors-allow-origins=http://127.0.0.1:7860" -# --opt-channelslast +CMD="launch.py --api --xformers --disable-console-progressbars --gradio-queue --skip-version-check --skip-install --skip-torch-cuda-test --disable-nan-check --cors-allow-origins=http://127.0.0.1:7860" + MODE=optimized if [[ $(id -u) -eq 0 ]]; then @@ -104,3 +102,7 @@ if [ $MODE == optimized ]; then fi exec accelerate launch --no_python --quiet --num_cpu_threads_per_process=6 "$PYTHON" $CMD + +# export LD_PRELOAD=libtcmalloc.so +# TORCH_CUDA_ARCH_LIST="8.6" +# --opt-channelslast diff --git a/modules/sd_models.py b/modules/sd_models.py index 93959f55f..2044bcbe5 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -278,7 +278,7 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer if depth_model: model.depth_model = depth_model - timer.record("apply half()") + timer.record("apply half") devices.dtype = torch.float32 if shared.cmd_opts.no_half else torch.float16 devices.dtype_vae = torch.float32 if shared.cmd_opts.no_half or shared.cmd_opts.no_half_vae else torch.float16 diff --git a/ui-config.json b/ui-config.json index ece495378..bda8feb9e 100644 --- a/ui-config.json +++ b/ui-config.json @@ -1169,5 +1169,9 @@ "train/Bake in VAE/visible": true, "train/Bake in VAE/value": "None", "train/Discard weights with matching name/visible": true, - "train/Discard weights with matching name/value": "" + "train/Discard weights with matching name/value": "", + "customscript/loopback.py/img2img/Append interrogated prompt at each iteration/visible": true, + "customscript/loopback.py/img2img/Append interrogated prompt at each iteration/value": "None", + "train/Use PNG alpha channel as loss weight/visible": true, + "train/Use PNG alpha channel as loss weight/value": false } \ No newline at end of file