JarvisIR/package/agent_tools/LightenDiffusion/utils/sampling.py

import torch
# This script is adapted from the following repository: https://github.com/ermongroup/ddim


def data_transform(X):
    return 2 * X - 1.0


def inverse_data_transform(X):
    return torch.clamp((X + 1.0) / 2.0, 0.0, 1.0)