Need transformer config for inference

#1
by meets2tarun - opened

Can I get the config as below for the inferencing
Config: {"dim": 768, "depth": 18, "heads": 12, "ff_mult": 2, "text_dim": 512, "text_mask_padding": False, "conv_layers": 4, "pe_attn_head": 1}

Thank you

Sign up or log in to comment