feat: add noise parameter

hydra_node/models.py

@@ -223,7 +223,7 @@ class StableDiffusionModel(nn.Module):
                 print("init latent shape:")
                 print(init_latent.shape)
 
-                init_latent = init_latent + (torch.randn_like(init_latent) * 0.667)
+                init_latent = init_latent + (torch.randn_like(init_latent) * request.noise)
 
                 uc = None
                 if request.scale != 1.0:

hydra_node/sanitize.py

@@ -62,6 +62,9 @@ def sanitize_stable_diffusion(request):
     if request.strength < 0.0 or request.strength > 1.0:
         return False, "strength should be more than 0.0 and less than 1.0"
 
+    if request.noise < 0.0 or request.noise > 1.0:
+        return False, "noise should be more than 0.0 and less than 1.0"
+
     return True, request
 
 def sanitize_dalle_mini(request):

main.py

@@ -69,6 +69,7 @@ class GenerationRequest(BaseModel):
     grid_size: int = 4
     advanced: bool = False
     strength: float = 0.69
+    noise: float = 0.667
 
 class GenerationOutput(BaseModel):
     output: List[str]