Blame for diffusion_core/utils/model_utils.py

import torch


def use_deterministic():
    torch.backends.cudnn.benchmark = False
    torch.backends.cudnn.deterministic = True


def toggle_grad(model, mode=True):
    for p in model.parameters():
        p.requires_grad = mode