Merge pull request #17 from neph1/update-v0.10.1

remove duplicate settings

Author: neph1

config/config_template.yaml

@@ -11,7 +11,7 @@ checkpointing_steps: 500
 data_root: ''
 dataloader_num_workers: 0
 dataset_file: ''
-diffusion_options: ''
+diffusion_options: '--flow_weighting_scheme logit_normal'
 enable_model_cpu_offload: false
 enable_slicing: true
 enable_tiling: true
@@ -30,7 +30,7 @@ lr_num_cycles: 1
 lr_scheduler: ['linear', 'cosine', 'cosine_with_restarts', 'polynomial', 'constant', 'constant_with_warmup']
 lr_warmup_steps: 400
 max_grad_norm: 1.0
-model_name: ltx_video
+model_name: ['ltx_video', 'hunyuan_video']
 nccl_timeout: 1800
 num_validation_videos: 0
 optimizer: adamw

run_trainer.py

@@ -24,6 +24,7 @@ def run(self, config: Config, finetrainers_path: str, log_file: str):
                      "--text_encoder_dtype", config.get('text_encoder_dtype'),
                      "--text_encoder_2_dtype", config.get('text_encoder_2_dtype'),
                      "--text_encoder_3_dtype", config.get('text_encoder_3_dtype'),
+                     "--transformer_dtype", config.get('transformer_dtype'),
                      "--vae_dtype", config.get('vae_dtype')]
 
         if config.get('layerwise_upcasting_modules') != 'none':
@@ -41,18 +42,12 @@ def run(self, config: Config, finetrainers_path: str, log_file: str):
         dataset_cmd += config.get('image_resolution_buckets').split(' ')
         dataset_cmd += ["--caption_dropout_p", config.get('caption_dropout_p'),
                    "--caption_dropout_technique", config.get('caption_dropout_technique'),
-                   "--text_encoder_dtype", config.get('text_encoder_dtype'),
-                   "--text_encoder_2_dtype", config.get('text_encoder_2_dtype'),
-                   "--text_encoder_3_dtype", config.get('text_encoder_3_dtype'),
-                   "--vae_dtype", config.get('vae_dtype'),
-                   "--transformer_dtype", config.get('transformer_dtype'),
                    '--precompute_conditions' if config.get('precompute_conditions') else '']
         if config.get('dataset_file'):
             dataset_cmd += ["--dataset_file", config.get('dataset_file')]
 
         dataloader_cmd = ["--dataloader_num_workers", config.get('dataloader_num_workers')]
 
-        # Diffusion arguments TODO: replace later
         diffusion_cmd = [config.get('diffusion_options')]
 
         training_cmd = ["--training_type", config.get('training_type'),