| acceleration: |
| load_text_encoder_in_8bit: false |
| mixed_precision_mode: bf16 |
| quantization: null |
| checkpoints: |
| interval: 100 |
| keep_last_n: -1 |
| precision: bfloat16 |
| data: |
| num_dataloader_workers: 2 |
| preprocessed_data_root: .precomputed |
| flow_matching: |
| timestep_sampling_mode: power_law |
| timestep_sampling_params: {} |
| hub: |
| hub_model_id: null |
| push_to_hub: false |
| lora: |
| alpha: 128 |
| dropout: 0.0 |
| rank: 128 |
| target_modules: |
| - attn1.to_k |
| - attn1.to_q |
| - attn1.to_v |
| - attn1.to_out.0 |
| - attn2.to_k |
| - attn2.to_q |
| - attn2.to_v |
| - attn2.to_out.0 |
| - ff.net.0.proj |
| - ff.net.2 |
| - adaln_single.linear |
| model: |
| load_checkpoint: /workspace/LTX-2/packages/ltx-trainer/new_simple_dataset_adamw_new_dataset_final_test_ga1/ckpt/lora_weights_step_01400.safetensors |
| model_path: /workspace/models/ltx-2-19b-dev.safetensors |
| text_encoder_path: /workspace/models/gemma |
| training_mode: lora |
| optimization: |
| batch_size: 1 |
| enable_gradient_checkpointing: true |
| gradient_accumulation_steps: 1 |
| learning_rate: 0.0001 |
| max_grad_norm: 1.0 |
| optimizer_type: adamw |
| scheduler_params: {} |
| scheduler_type: cosine |
| steps: 2000 |
| output_dir: /workspace/LTX-2/packages/ltx-trainer/outputs/head_swap |
| seed: 42 |
| training_strategy: |
| background_loss_weight: 0.1 |
| first_frame_conditioning_p: 1.0 |
| mask_dir: masks |
| mask_loss_weight: 1.0 |
| name: avatar_audio |
| reference_latents_dir: reference_latents |
| use_masked_loss: true |
| validation: |
| frame_rate: 25.0 |
| generate_audio: false |
| guidance_scale: 3.0 |
| images: |
| - /workspace/LTX-2/packages/ltx-trainer/dataset/test/face.png |
| include_reference_in_output: true |
| inference_steps: 30 |
| interval: 100 |
| negative_prompt: worst quality, inconsistent motion, blurry, jittery, distorted |
| prompts: |
| - head_swap |
| reference_downscale_factor: 1 |
| reference_videos: |
| - /workspace/LTX-2/packages/ltx-trainer/dataset/test/guide.mp4 |
| seed: 42 |
| skip_initial_validation: true |
| stg_blocks: |
| - 29 |
| stg_mode: stg_av |
| stg_scale: 1.0 |
| video_dims: !!python/tuple |
| - 416 |
| - 768 |
| - 121 |
| videos_per_prompt: 1 |
| wandb: |
| enabled: true |
| entity: null |
| log_validation_videos: true |
| project: head_swap |
| tags: |
| - ltx2 |
| - head_swap |
|
|
