This commit is contained in:
Vladimir Mandic
2023-02-19 11:03:11 -05:00
parent 2247f0c367
commit fc3e9e5d5b
3 changed files with 12 additions and 6 deletions
+6 -4
View File
@@ -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
+1 -1
View File
@@ -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
+5 -1
View File
@@ -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
}