Spaces:

heheyas
/

V3D

Running on Zero

V3D / sgm /modules /diffusionmodules /denoiser_weighting.py

heheyas

init

cfb7702 3 months ago

No virus

516 Bytes

	import torch


	class UnitWeighting:
	def __call__(self, sigma):
	return torch.ones_like(sigma, device=sigma.device)


	class EDMWeighting:
	def __init__(self, sigma_data=0.5):
	self.sigma_data = sigma_data

	def __call__(self, sigma):
	return (sigma2 + self.sigma_data2) / (sigma * self.sigma_data) ** 2


	class VWeighting(EDMWeighting):
	def __init__(self):
	super().__init__(sigma_data=1.0)


	class EpsWeighting:
	def __call__(self, sigma):
	return sigma**-2.0