amd
/

hecui102 commited on
Commit
72e7aed
·
verified ·
1 Parent(s): 1e9bde1

Delete run.sh

Browse files
Files changed (1) hide show
  1. run.sh +0 -51
run.sh DELETED
@@ -1,51 +0,0 @@
1
- version=$1 ## 1024, 512, 256
2
- GPU=3
3
- ckpt='./stage_1.ckpt'
4
- config='configs/inference_512_v1.0_07B.yaml'
5
- prompt_dir="/VBench"
6
- base_res_dir="result"
7
-
8
- # 判断分辨率
9
- if [ "$1" == "256" ]; then
10
- H=256
11
- FS=3
12
- elif [ "$1" == "512" ]; then
13
- H=320
14
- FS=24
15
- elif [ "$1" == "1024" ]; then
16
- H=576
17
- FS=10
18
- else
19
- echo "Invalid input. Please enter 256, 512, or 1024."
20
- exit 1
21
- fi
22
-
23
-
24
- seed='123'
25
- sub_dir='0'
26
- res_dir="${base_res_dir}/${sub_dir}"
27
-
28
- echo "Running seed=$seed -> Saving to: $res_dir"
29
- echo $prompt_dir
30
-
31
- CUDA_VISIBLE_DEVICES=$GPU python3 scripts/evaluation/inference.py \
32
- --seed ${seed} \
33
- --ckpt_path $ckpt \
34
- --config $config \
35
- --savedir $res_dir \
36
- --n_samples 1 \
37
- --bs 1 --height ${H} --width $1 \
38
- --unconditional_guidance_scale 7.5 \
39
- --ddim_steps 16 \
40
- --ddim_eta 1.0 \
41
- --prompt_dir $prompt_dir \
42
- --text_input \
43
- --video_length 16 \
44
- --frame_stride ${FS} \
45
- --use_unet 1 \
46
- --unet_path 'stae_2/output/unet.pt' \
47
- --img_proj_path 'stage_2/output/img_proj.pt' \
48
- --timestep_spacing 'uniform_trailing' \
49
- --guidance_rescale 0.7 \
50
- --perframe_ae
51
-