Upload 7 files

Cradle2Cane/clip_project.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf40e6d790f1e664358e5a5261e3dace0d9e9cf7d12c2096f73e365f8c724f9e
+size 73582438

Cradle2Cane/face_project.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c1da7718fec9faf580d23cd2b408f25b932dc3d6dbe6460a7895faafca2bd52b
+size 1695160934

Cradle2Cane/hparams.yml

@@ -0,0 +1,104 @@
+config: config/train_config.yaml
+pretrained_model_name_or_path: ./models/sdxl-turbo
+pretrained_vae_model_name_or_path: ./models/sdxl-vae-fp16-fix
+dataset_name: config/ffhq_dataset.yaml
+output_dir: exp/ablation_20000_ep30
+seed: 42
+num_train_epochs: 30
+train_batch_size: 5
+validation_steps: 1000
+max_load_num: 20000
+max_train_steps: 20010
+checkpointing_steps: 4000
+t1: 10
+t2: 30
+id_cos_loss_weight: 1.0
+age_loss_weight: 0.75
+age_loss_2_weight: 1.0
+pixel_mse_loss_weight: 0.05
+ssim_loss_weight: 0.1
+g_loss_weight: 0.01
+lpips_loss_weight: 1.0
+arcface_weight: ./models/backbone.pth
+clip_L_path: ./models/clip-vit-large-patch14/
+lpips_model_path: ./models/alex.pth
+age_model_path: ./models/model_only_age_imdb_4.29.pth.tar
+train_data_list: ./models/ffhq_crop_gray_wo_crop_race4
+lpips_net: alex
+arcface_network: r100
+revision: null
+resolution: 512
+center_crop: false
+random_flip: true
+learning_rate: 0.0001
+scale_lr: false
+lr_scheduler: constant
+lr_warmup_steps: 500
+gradient_accumulation_steps: 1
+gradient_checkpointing: true
+mixed_precision: fp16
+variant: fp16
+start_g_loss_weigth: 0.04
+end_g_loss_weigth: 0.04
+d_loss_weight: 1.0
+d_loss_update_step: 6
+min_age: 1
+max_age: 80
+same_age_rate: 0.01
+age_scale: 20
+aged_strength: 0.5
+id_strength: 0.25
+infer_step: 4
+gender_strength: 0.5
+use_avg: true
+sort: false
+use_four_image_swr: false
+use_swr: false
+use_gan_loss: true
+use_arcface_project: true
+use_clip_project: true
+face_project_use_hidden_state: true
+pixel_mse_loss_use_weight: true
+lpips_loss_use_weight: false
+use_adaptive_noise_inject: true
+prompt_mode: normal
+age_sample_mode: uniform
+clip_map_model: angle
+rank: 40
+use_8bit_adam: false
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_weight_decay: 0.01
+adam_epsilon: 1.0e-08
+max_grad_norm: 1.0
+train_text_encoder: false
+snr_gamma: null
+prediction_type: null
+enable_xformers_memory_efficient_attention: true
+noise_offset: 0
+allow_tf32: false
+dataloader_num_workers: 3
+checkpoints_total_limit: null
+resume_from_checkpoint: null
+num_validation_images: 1
+push_to_hub: false
+hub_token: null
+hub_model_id: null
+logging_dir: logs
+report_to: tensorboard
+debug_loss: false
+debug_index: -1
+embedding_fuse_save_name: embedding_fuse.bin
+face_project_save_name: face_project.bin
+face_project_hidden_dim: 25088
+face_project_feature_dim: 512
+face_project_output_dim: 2048
+clip_project_save_name: clip_project.bin
+clip_project_input_dim: 768
+clip_project_output_dim: 2048
+swr_alpha: 0.01
+swr_beta: 1.1
+local_rank: 0
+_config_loaded_at: '2025-09-25 11:01:57'
+_config_loaded_from: train_config.yaml
+_config_loaded_by: u2120240694

Cradle2Cane/pytorch_lora_weights.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:224380cbd5e17b3fb25e17c3b93c2580b68137f738e31ff4cc7e07a41ffd7411
+size 232412664

alex.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:df73285e35b22355a2df87cdb6b70b343713b667eddbda73e1977e0c860835c0
+size 6009

backbone.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a566a62357f0c55b679d9ff2f022a294486568be0c00665d39029d0e46a8109b
+size 261223796

model_only_age_imdb_4.29.pth.tar

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:53dedc7d173f61e6b0a14de4195aa051b98b3d3b76480d67add3ef7b226c30b0
+size 103492590