You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
NeMo's StableDiffusion uses SiLU activation in its UNet model. When UNet stage is run with thunder.jit, thunder gives an error: silu() got an unexpected keyword argument 'inplace'.
Specifically the error is is raised in this line of torch/nn/modules/activation.py although inplace is a valid argument of the op.
we can add the inplace argument to the signature and throw a NotImplementedError if it's True for now, that will also provide a more informative error message to tell us if this is actually being used inplace
we should revisit general inplace operator support with Q3 planning
🐛 Bug
NeMo's StableDiffusion uses SiLU activation in its UNet model. When
UNet
stage is run withthunder.jit
, thunder gives an error:silu() got an unexpected keyword argument 'inplace'
.Specifically the error is is raised in this line of
torch/nn/modules/activation.py
althoughinplace
is a valid argument of the op.CC: @tfogal
cc @tfogal
The text was updated successfully, but these errors were encountered: