Commit 90b418ca authored by kurumuz's avatar kurumuz

fix seeding for img2img

parent ae820236
...@@ -177,6 +177,10 @@ class StableDiffusionModel(nn.Module): ...@@ -177,6 +177,10 @@ class StableDiffusionModel(nn.Module):
@torch.no_grad() @torch.no_grad()
@torch.autocast("cuda", enabled=True, dtype=torch.float16) @torch.autocast("cuda", enabled=True, dtype=torch.float16)
def sample(self, request): def sample(self, request):
if request.seed is not None:
torch.manual_seed(request.seed)
np.random.seed(request.seed)
if request.image is not None: if request.image is not None:
request.sampler = "ddim_img2img" #enforce ddim for now request.sampler = "ddim_img2img" #enforce ddim for now
self.ddim.make_schedule(ddim_num_steps=request.steps, ddim_eta=request.ddim_eta, verbose=False) self.ddim.make_schedule(ddim_num_steps=request.steps, ddim_eta=request.ddim_eta, verbose=False)
...@@ -185,10 +189,6 @@ class StableDiffusionModel(nn.Module): ...@@ -185,10 +189,6 @@ class StableDiffusionModel(nn.Module):
print(start_code.shape) print(start_code.shape)
start_code = start_code + (torch.randn_like(start_code) * request.noise) start_code = start_code + (torch.randn_like(start_code) * request.noise)
t_enc = int(request.strength * request.steps) t_enc = int(request.strength * request.steps)
if request.seed is not None:
torch.manual_seed(request.seed)
np.random.seed(request.seed)
if request.sampler.startswith("k_"): if request.sampler.startswith("k_"):
sampler = "k-diffusion" sampler = "k-diffusion"
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment