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
|
@register_model
|
||||||
def register_transformer_diffusion4(opt_net, opt):
|
def register_transformer_diffusion5(opt_net, opt):
|
||||||
return TransformerDiffusion(**opt_net['kwargs'])
|
return TransformerDiffusion(**opt_net['kwargs'])
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user