allow generation to be started with any dimensions specified

This commit is contained in:
AUTOMATIC1111
2024-06-28 18:06:49 +03:00
parent 179ae47d64
commit d67348a0a5
2 changed files with 6 additions and 0 deletions

View File

@@ -75,3 +75,6 @@ class SD3Inferencer(torch.nn.Module):
def add_noise_to_latent(self, x, noise, amount):
return x * (1 - amount) + noise * amount
def fix_dimensions(self, width, height):
return width // 16 * 16, height // 16 * 16