In the file model_jit.py, there is no configuration for jit G. Is my configuration below correct?
def JiT_G_32(**kwargs):
return JiT(
depth=40, hidden_size=1664, num_heads=16,
bottleneck_dim=256, in_context_len=32, in_context_start=10,
patch_size=32, **kwargs
)