forked from mrq/DL-Art-School
df5
This commit is contained in:
parent
6b43915eb8
commit
da367da411
|
@ -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'])
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user