diff --git a/labml_nn/diffusion/stable_diffusion/model/unet.py b/labml_nn/diffusion/stable_diffusion/model/unet.py index 761bded9856b1fc4060e40ec28fc6f3fef650dbe..261a4bcedf0b59525fcc06d928701c758c7445ea 100644 --- a/labml_nn/diffusion/stable_diffusion/model/unet.py +++ b/labml_nn/diffusion/stable_diffusion/model/unet.py @@ -186,7 +186,7 @@ class TimestepEmbedSequential(nn.Sequential): """ ### Sequential block for modules with different inputs - This sequential module can compose of different modules suck as `ResBlock`, + This sequential module can compose of different modules such as `ResBlock`, `nn.Conv` and `SpatialTransformer` and calls them with the matching signatures """