-
Notifications
You must be signed in to change notification settings - Fork 256
/
Wonder3D_config.yaml
42 lines (34 loc) · 1012 Bytes
/
Wonder3D_config.yaml
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
34
35
36
37
38
39
40
41
42
#pretrained_model_name_or_path: 'flamehaze1115/wonder3d-v1.0' # or './ckpts'
revision: null
validation_dataset:
root_dir: "./example_images" # the folder path stores testing images
num_views: 6
bg_color: 'white'
img_wh: [256, 256]
num_validation_samples: 1000
crop_size: 192
filepaths: ['owl.png'] # the test image names. leave it empty, test all images in the folder
save_dir: 'outputs/'
pred_type: 'joint'
seed: 42
validation_batch_size: 1
dataloader_num_workers: 64
local_rank: -1
pipe_kwargs:
camera_embedding_type: 'e_de_da_sincos'
num_views: 6
validation_guidance_scales: [1.0]
pipe_validation_kwargs:
eta: 1.0
validation_grid_nrow: 6
unet_from_pretrained_kwargs:
camera_embedding_type: 'e_de_da_sincos'
projection_class_embeddings_input_dim: 10
num_views: 6
sample_size: 32
cd_attention_mid: true
zero_init_conv_in: false
zero_init_camera_projection: false
num_views: 6
camera_embedding_type: 'e_de_da_sincos'
enable_xformers_memory_efficient_attention: true