wangfuyun commited on
Commit
cf6765f
1 Parent(s): 6af3ee3

Upload inference.yaml

Browse files
Files changed (1) hide show
  1. configs/inference.yaml +31 -0
configs/inference.yaml ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ unet_additional_kwargs:
2
+ use_pixel_encoder: false
3
+ use_img_encoder: false
4
+ use_inflated_groupnorm: true
5
+ unet_use_cross_frame_attention: false
6
+ unet_use_temporal_attention: false
7
+ use_motion_module: true
8
+ use_motion_resnet: false
9
+ motion_module_resolutions:
10
+ - 1
11
+ - 2
12
+ - 4
13
+ - 8
14
+ motion_module_mid_block: true
15
+ motion_module_decoder_only: false
16
+ motion_module_type: Vanilla
17
+ motion_module_kwargs:
18
+ num_attention_heads: 8
19
+ num_transformer_block: 1
20
+ attention_block_types:
21
+ - Temporal_Self
22
+ - Temporal_Self
23
+ temporal_position_encoding: true
24
+ temporal_attention_dim_div: 1
25
+
26
+ noise_scheduler_kwargs:
27
+ beta_start: 0.00085
28
+ beta_end: 0.012
29
+ beta_schedule: "linear"
30
+ original_inference_steps: 100
31
+ steps_offset: 1