Imported from https://github.com/charlescheng0117/pytorch-VideoVAE