flux.1-kontext-dev.yaml

pipeline:
  name: flux.1-kontext-dev
  path: svdq/flux.1-kontext-dev
  dtype: torch.bfloat16
eval:
  num_steps: 8
  guidance_scale: 1
  protocol: fmeuler{num_steps}-g{guidance_scale}
quant:
  calib:
    batch_size: 16
  wgts:
    calib_range:
      element_batch_size: 16
      sample_batch_size: 4
      element_size: 128
      sample_size: -1
    low_rank:
      sample_batch_size: 4
      sample_size: -1
    skips: 
    - embed
    - resblock_shortcut
    - resblock_time_proj
    - transformer_proj_in
    - transformer_proj_out
    - down_sample
    - up_sample
  ipts:
    calib_range:
      element_batch_size: 16
      sample_batch_size: 4
      element_size: 128
      sample_size: -1
    skips: 
    - embed
    - resblock_shortcut
    - resblock_time_proj
    - transformer_proj_in
    - transformer_proj_out
    - transformer_norm
    - transformer_add_norm
    - down_sample
    - up_sample
  opts:
    calib_range:
      element_batch_size: 16
      sample_batch_size: 4
      element_size: 128
      sample_size: -1
  smooth:
    proj:
      element_batch_size: -1
      sample_batch_size: 4
      element_size: -1
      sample_size: -1
    attn:
      sample_batch_size: 4
      sample_size: -1