-
Notifications
You must be signed in to change notification settings - Fork 11
/
inference_custom_audio_show.sh
33 lines (31 loc) · 1.09 KB
/
inference_custom_audio_show.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# For 50+ FPS on A100
PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
OMP_NUM_THREADS=10 CUDA_VISIBLE_DEVICES=0 python -u runner.py \
--dataset_name talkshow \
--name talkshow_GesExpr_unify_addHubert_encodeHubert_mdlpIncludeX_condRes_LN_ClsFree \
--n_poses 88 \
--model_base transformer_encoder \
--classifier_free \
--cond_scale 1.15 \
--ckpt ckpt_e2599.tar \
--ddim \
--timestep_respacing ddim25 \
--overlap_len 10 \
--mode test_custom_audio \
--test_audio_path audios/Forrest_tts.wav
## For 120+ FPS on A100
# PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
# OMP_NUM_THREADS=10 CUDA_VISIBLE_DEVICES=0 python -u runner.py \
# --dataset_name talkshow \
# --name talkshow_GesExpr_unify_addHubert_encodeHubert_mdlpIncludeX_condRes_LN_ClsFree \
# --n_poses 88 \
# --model_base transformer_encoder \
# --classifier_free \
# --cond_scale 1.15 \
# --ckpt ckpt_e2599.tar \
# --ddim \
# --timestep_respacing ddim25 \
# --overlap_len 10 \
# --mode test_custom_audio \
# --jump_n_sample 2 \
# --test_audio_path audios/Forrest_tts.wav