sky_sample.yaml 574 Bytes
Newer Older
mashun1's avatar
latte  
mashun1 committed
1
2
3
# path:
ckpt: # will be overwrite
save_img_path: "./sample_videos/" # will be overwrite
mashun1's avatar
mashun1 committed
4
pretrained_model_path: "pretrained_models"
mashun1's avatar
latte  
mashun1 committed
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

# model config: 
model: Latte-XL/2
num_frames: 16
image_size: 256 # choices=[256, 512]
frame_interval: 2
fixed_spatial: False
attention_bias: True
learn_sigma: True
extras: 1 # [1, 2, 78]
num_classes:

# model speedup
use_compile: False
use_fp16: True

# sample config:
seed:
sample_method: 'ddpm'
num_sampling_steps: 250
cfg_scale: 1.0
run_time: 12
num_sample: 1
negative_name:

# ddp sample config
per_proc_batch_size: 1
mashun1's avatar
mashun1 committed
32
num_fvd_samples: 2