VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix sample image generation without seed failed close #1726

Browse Source
This commit is contained in:
kohya-ss
2024-10-26 18:59:45 +09:00
parent 56bf761164
commit 014064fd81
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/sd3_train_utils.py
View File

@@ -316,6 +316,8 @@ def do_sample(
# noise = get_noise(seed, latent).to(device) # noise = get_noise(seed, latent).to(device)
if seed is not None: if seed is not None:
generator = torch.manual_seed(seed) generator = torch.manual_seed(seed)
else:
generator = None
noise = ( noise = (
torch.randn(latent.size(), dtype=torch.float32, layout=latent.layout, generator=generator, device="cpu") torch.randn(latent.size(), dtype=torch.float32, layout=latent.layout, generator=generator, device="cpu")
.to(latent.dtype) .to(latent.dtype)