|
|
@@ -136,13 +136,13 @@ export PYTHON_BIN_RESTART=tmp/restart
|
|
|
while [[ "$KEEP_GOING" -eq "1" ]]; do
|
|
|
if [[ ! -z "${ACCELERATE}" ]] && [ ${ACCELERATE}="True" ] && [ -x "$(command -v accelerate)" ]; then
|
|
|
printf "\n%s\n" "${delimiter}"
|
|
|
- printf "Accelerating launch.py..."
|
|
|
+ printf "Accelerating wenda.py..."
|
|
|
printf "\n%s\n" "${delimiter}"
|
|
|
prepare_tcmalloc
|
|
|
accelerate launch --num_cpu_threads_per_process=6 "${LAUNCH_SCRIPT}" -t rwkv "$@"
|
|
|
else
|
|
|
printf "\n%s\n" "${delimiter}"
|
|
|
- printf "Launching launch.py..."
|
|
|
+ printf "Launching wenda.py..."
|
|
|
printf "\n%s\n" "${delimiter}"
|
|
|
prepare_tcmalloc
|
|
|
"${python_cmd}" -u "${LAUNCH_SCRIPT}" -t rwkv "$@"
|