This commit is contained in:
James Betker 2022-05-28 22:30:23 -06:00
parent 6b43915eb8
commit da367da411

View File

@ -196,7 +196,7 @@ class TransformerDiffusion(nn.Module):
@register_model
def register_transformer_diffusion4(opt_net, opt):
def register_transformer_diffusion5(opt_net, opt):
return TransformerDiffusion(**opt_net['kwargs'])