mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
update
This commit is contained in:
+6
-4
@@ -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
|
||||
|
||||
@@ -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
@@ -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
|
||||
}
|
||||
Reference in New Issue
Block a user