code: def img2img_transformer2d_tiny():
    
    hparams = img2img_transformer2d_base()
    hparams.num_decoder_layers = 2
    hparams.hidden_size = 128
    hparams.batch_size = 4
    hparams.max_length = 128
    hparams.attention_key_channels = hparams.attention_value_channels = 0
    hparams.filter_size = 128
    hparams.num_heads = 4
    hparams.pos = "timing"
    hparams.img_len = 32
    return hparams
docs: Tiny params.