code: def img2img_transformer_dilated():
    
    hparams = img2img_transformer_base()
    hparams.add_hparam("num_memory_blocks", 1)
    hparams.num_heads = 8
    hparams.attention_key_channels = hparams.attention_value_channels = 0
    hparams.hidden_size = 512
    hparams.filter_size = 2048
    hparams.num_decoder_layers = 8
    hparams.sampling_method = "random"
    hparams.gap_sizes = [0, 16, 64, 0, 16, 64, 128, 0]
    hparams.dec_attention_type = cia.AttentionType.DILATED
    hparams.img_len = 64
    hparams.block_length = 128
    hparams.block_width = 128
    return hparams
docs: Try dilated.