File size: 1,637 Bytes
b213d84 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
# File used for running pure prediction with torchtnt, including a custom score computation.
seed: 42
defaults:
- constants: base
# - model: idm_vton_sd15
# - model: idm_vton_sdxl
# - model: cat_vton_sd15
# - model: cat_vton_sdxl
- model: simple_vton_sd15
# - model: simple_vton_sdxl
# - model: simple_vton_sd15_fluxvae
# - model: simple_vton_sdxl_fluxvae
- datasets:
- viton_hd_test
- dress_code_test
- deepfashion_test
- _self_
constants:
# for virtual try-on
# height: 512
# width: 384
height: 1024
width: 768
# for pose transfer
# height: 256
# width: 176
# height: 512
# width: 352
# height: 1024
# width: 704
batch_size: 1
# null to go through whole dataloader.
max_steps_per_epoch: null
dataloader: ${datasets.viton_hd_test.dataloader}
# dataloader: ${datasets.dress_code_test.dataloader}
# dataloader: ${datasets.deepfashion_test.dataloader}
manifold_log_dir: manifold://genads_models/tree/zijianzhou/output/simple_vton/tmp
inference:
# _target_: leffa.inference.IdmVtonInference
# _target_: leffa.inference.CatVtonInference
_target_: leffa.inference.SimpleVtonInference
_partial_: True
model: ${model}
model_entity_id: null
checkpoint_version: null
callbacks:
- _target_: leffa.callbacks.save_image_callback.SaveImageCallback
manifold_path: ${manifold_log_dir}/generated_paired
# manifold_path: ${manifold_log_dir}/generated_unpaired
task_type: vton
# manifold_path: ${manifold_log_dir}
# task_type: pose_transfer
unit:
_target_: leffa.vton_pred_unit.VtonPredUnit
inference_fn: ${inference}
hydra:
run:
dir: /tmp/
|