From b63a9928876157e2f5b4cc550c29448470870f9e Mon Sep 17 00:00:00 2001 From: FeepingCreature Date: Fri, 30 Aug 2024 15:25:05 +0200 Subject: [PATCH] Fix seed assignment for batch requests. --- ai_diffusion/model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ai_diffusion/model.py b/ai_diffusion/model.py index 3736e4e78..eb2359a43 100644 --- a/ai_diffusion/model.py +++ b/ai_diffusion/model.py @@ -1,6 +1,7 @@ from __future__ import annotations import asyncio from copy import copy +from dataclasses import replace from pathlib import Path from enum import Enum from typing import Any, NamedTuple @@ -209,8 +210,8 @@ async def enqueue_jobs( params.prompt = params.regions[0].prompt for i in range(count): - sampling.seed = sampling.seed + i * settings.batch_size - params.seed = sampling.seed + input = replace(input, sampling=replace(sampling, seed=sampling.seed + i * settings.batch_size)) + params.seed = input.sampling.seed job = self.jobs.add(kind, copy(params)) await self._enqueue_job(job, input)