ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 03:01:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix seed generation to make it deterministic so it is consistant from gpu to gpu

Browse Source
This commit is contained in:
Jaret Burkett
2024-11-15 12:11:13 -07:00
parent 025ee3dd3d
commit 6509ba4484
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
toolkit/stable_diffusion_model.py
View File

@@ -1161,6 +1161,8 @@ class StableDiffusion:
self.network.multiplier = gen_config.network_multiplier self.network.multiplier = gen_config.network_multiplier
torch.manual_seed(gen_config.seed) torch.manual_seed(gen_config.seed)
torch.cuda.manual_seed(gen_config.seed) torch.cuda.manual_seed(gen_config.seed)
generator = torch.manual_seed(gen_config.seed)
if self.adapter is not None and isinstance(self.adapter, ClipVisionAdapter) \ if self.adapter is not None and isinstance(self.adapter, ClipVisionAdapter) \
and gen_config.adapter_image_path is not None: and gen_config.adapter_image_path is not None:
@@ -1285,6 +1287,7 @@ class StableDiffusion:
guidance_scale=gen_config.guidance_scale, guidance_scale=gen_config.guidance_scale,
guidance_rescale=grs, guidance_rescale=grs,
latents=gen_config.latents, latents=gen_config.latents,
generator=generator,
**extra **extra
).images[0] ).images[0]
elif self.is_v3: elif self.is_v3:
@@ -1298,6 +1301,7 @@ class StableDiffusion:
num_inference_steps=gen_config.num_inference_steps, num_inference_steps=gen_config.num_inference_steps,
guidance_scale=gen_config.guidance_scale, guidance_scale=gen_config.guidance_scale,
latents=gen_config.latents, latents=gen_config.latents,
generator=generator,
**extra **extra
).images[0] ).images[0]
elif self.is_flux: elif self.is_flux:
@@ -1312,6 +1316,7 @@ class StableDiffusion:
num_inference_steps=gen_config.num_inference_steps, num_inference_steps=gen_config.num_inference_steps,
guidance_scale=gen_config.guidance_scale, guidance_scale=gen_config.guidance_scale,
latents=gen_config.latents, latents=gen_config.latents,
generator=generator,
**extra **extra
).images[0] ).images[0]
else: else:
@@ -1325,6 +1330,7 @@ class StableDiffusion:
num_inference_steps=gen_config.num_inference_steps, num_inference_steps=gen_config.num_inference_steps,
guidance_scale=gen_config.guidance_scale, guidance_scale=gen_config.guidance_scale,
latents=gen_config.latents, latents=gen_config.latents,
generator=generator,
**extra **extra
).images[0] ).images[0]
elif self.is_pixart: elif self.is_pixart:
@@ -1345,6 +1351,7 @@ class StableDiffusion:
num_inference_steps=gen_config.num_inference_steps, num_inference_steps=gen_config.num_inference_steps,
guidance_scale=gen_config.guidance_scale, guidance_scale=gen_config.guidance_scale,
latents=gen_config.latents, latents=gen_config.latents,
generator=generator,
**extra **extra
).images[0] ).images[0]
elif self.is_auraflow: elif self.is_auraflow:
@@ -1366,6 +1373,7 @@ class StableDiffusion:
num_inference_steps=gen_config.num_inference_steps, num_inference_steps=gen_config.num_inference_steps,
guidance_scale=gen_config.guidance_scale, guidance_scale=gen_config.guidance_scale,
latents=gen_config.latents, latents=gen_config.latents,
generator=generator,
**extra **extra
).images[0] ).images[0]
else: else:
@@ -1379,6 +1387,7 @@ class StableDiffusion:
num_inference_steps=gen_config.num_inference_steps, num_inference_steps=gen_config.num_inference_steps,
guidance_scale=gen_config.guidance_scale, guidance_scale=gen_config.guidance_scale,
latents=gen_config.latents, latents=gen_config.latents,
generator=generator,
**extra **extra
).images[0] ).images[0]
@@ -1404,7 +1413,8 @@ class StableDiffusion:
guidance_rescale=grs, guidance_rescale=grs,
denoising_start=gen_config.refiner_start_at, denoising_start=gen_config.refiner_start_at,
denoising_end=gen_config.num_inference_steps, denoising_end=gen_config.num_inference_steps,
image=img.unsqueeze(0) image=img.unsqueeze(0),
generator=generator,
).images[0] ).images[0]
gen_config.save_image(img, i) gen_config.save_image(img, i)