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

fix to use zero for initial latent

Browse Source
This commit is contained in:
Kohya S
2024-06-24 23:12:48 +09:00
parent d53ea22b2a
commit 0fe4eafac9
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sd3_minimal_inference.py
View File

@@ -64,7 +64,8 @@ def do_sample(
device: str, device: str,
): ):
if initial_latent is None: if initial_latent is None:
latent = torch.ones(1, 16, height // 8, width // 8, device=device) * 0.0609 # latent = torch.ones(1, 16, height // 8, width // 8, device=device) * 0.0609
latent = torch.zeros(1, 16, height // 8, width // 8, device=device)
else: else:
latent = initial_latent latent = initial_latent