Method forward

sat/sgm/modules/diffusionmodules/openaimodel.py:490–491

(self, t)

        self.dim = dim

    def forward(self, t):
        return timestep_embedding(t, self.dim)


str_to_dtype = {"fp32": th.float32, "fp16": th.float16, "bf16": th.bfloat16}