pengcc1 commited on
Commit
b2ff030
·
verified ·
1 Parent(s): f47cded

Upload config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.yaml +92 -0
config.yaml ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ image_finetune: false
2
+ name: training
3
+ use_wandb: false
4
+ launcher: pytorch
5
+ output_dir: outputs/training-2024-09-26T06-56-37
6
+ pretrained_model_path: /mnt/petrelfs/chenpengcheng/Tune_animatediff/models/StableDiffusion
7
+ train_data:
8
+ csv_path: /mnt/petrelfs/chenpengcheng/AnimateDiff/ChronoMagic/caption/ChronoMagic_train.csv
9
+ video_folder: /mnt/petrelfs/chenpengcheng/AnimateDiff/ChronoMagic/video
10
+ sample_size: 256
11
+ sample_stride: 4
12
+ sample_n_frames: 16
13
+ validation_data:
14
+ prompts:
15
+ - Time-lapse of a rose's blooming process, displaying tight buds progressively unfurling
16
+ into fully open blossoms. The petals gradually separate and spread, revealing
17
+ the flower's inner structure in a forward sequence against a contrasting black
18
+ background.
19
+ - A drone view of celebration with Christma tree and fireworks, starry sky - background.
20
+ - Robot dancing in times square.
21
+ - Pacific coast, carmel by the sea ocean and waves.
22
+ num_inference_steps: 25
23
+ guidance_scale: 8.0
24
+ cfg_random_null_text: true
25
+ unet_checkpoint_path: /mnt/petrelfs/chenpengcheng/AnimateDiff/mm15/mm_sd_v15.ckpt
26
+ unet_additional_kwargs:
27
+ use_motion_module: true
28
+ motion_module_resolutions:
29
+ - 1
30
+ - 2
31
+ - 4
32
+ - 8
33
+ unet_use_cross_frame_attention: false
34
+ unet_use_temporal_attention: false
35
+ motion_module_type: Vanilla
36
+ motion_module_kwargs:
37
+ num_attention_heads: 8
38
+ num_transformer_block: 1
39
+ attention_block_types:
40
+ - Temporal_Self
41
+ - Temporal_Self
42
+ temporal_position_encoding: true
43
+ temporal_position_encoding_max_len: 24
44
+ temporal_attention_dim_div: 1
45
+ zero_initialize: true
46
+ ema_decay: 0.9999
47
+ noise_scheduler_kwargs:
48
+ num_train_timesteps: 1000
49
+ beta_start: 0.00085
50
+ beta_end: 0.012
51
+ beta_schedule: linear
52
+ steps_offset: 1
53
+ clip_sample: false
54
+ max_train_epoch: -1
55
+ max_train_steps: 10000
56
+ validation_steps: 5000
57
+ validation_steps_tuple:
58
+ - 0
59
+ - 2
60
+ - 10
61
+ - 20
62
+ - 30
63
+ - 5000
64
+ - 9999
65
+ learning_rate: 5.0e-05
66
+ scale_lr: false
67
+ lr_warmup_steps: 0
68
+ lr_scheduler: constant
69
+ trainable_modules:
70
+ - motion_modules.
71
+ num_workers: 32
72
+ train_batch_size: 2
73
+ adam_beta1: 0.9
74
+ adam_beta2: 0.999
75
+ adam_weight_decay: 0.01
76
+ adam_epsilon: 1.0e-08
77
+ max_grad_norm: 1.0
78
+ gradient_accumulation_steps: 1
79
+ gradient_checkpointing: false
80
+ checkpointing_epochs: -1
81
+ checkpointing_steps: 200
82
+ mixed_precision_training: true
83
+ enable_xformers_memory_efficient_attention: true
84
+ global_seed: 42
85
+ is_debug: false
86
+ local_rank: 0
87
+ global_rank: 0
88
+ num_processes: 2
89
+ is_main_process: true
90
+ seed: 42
91
+ folder_name: training-2024-09-26T06-56-37
92
+ cfg_random_null_text_ratio: 0.1