Spaces:

hugging-apps
/

inverse-and-edit

Sleeping

inverse-and-edit / diffusion_core /utils /model_utils.py

Upload folder using huggingface_hub

dc7f251 verified 15 days ago

230 Bytes

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

	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