Files
automatic/train.sh
T
2023-01-23 16:44:42 -05:00

10 lines
328 B
Bash
Executable File

#/bin/env bash
echo "Running without optimizations"
export TF_CPP_MIN_LOG_LEVEL=2
export FORCE_CUDA="1"
export ATTN_PRECISION=fp16
export PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.9,max_split_size_mb:512
python launch.py --api --disable-opt-split-attention --disable-console-progressbars "$@"
# --opt-channelslast