Kohya S. kohya-ss

kohya-ss / gradual_latent.py

Last active March 15, 2024 09:52 — forked from laksjdjf/gradual_latent.py

	import comfy
	from comfy.samplers import KSAMPLER
	import torch
	from torchvision.transforms.functional import gaussian_blur
	from comfy.k_diffusion.sampling import default_noise_sampler, get_ancestral_step, to_d, BrownianTreeNoiseSampler
	from tqdm.auto import trange


	@torch.no_grad()
	def sample_euler_ancestral(