class_docstr: A sequential module that passes timestep embeddings to the children that
  support it as an extra input.
class_docstr_tok: ["A","sequential","module","that","passes","timestep","embeddings","to","the","children","that","support","it","as","an","extra","input","."]
code: def forward(self, x, emb, context=None):
      for layer in self:
          if isinstance(layer, TimestepBlock):
              x = layer(x, emb)
          elif isinstance(layer, SpatialTransformer):
              x = layer(x, context)
          else:
              x = layer(x)
      return x
  
imports: import math
  from abc import abstractmethod
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  from ldm.modules.attention import SpatialTransformer
  from ldm.modules.diffusionmodules.util import avg_pool_nd, checkpoint, conv_nd, linear, normalization, timestep_embedding, zero_module
  from ldm.util import exists
label_desc: Use image node_id 1 for calling the TimestepEmbedSequential obj's underlying member method code with example usage: obj.forward(x, emb, context) and returns: x