Hotswap: Handle noise==None

replicate · Nov 27, 2024 · 19c4ebc · 19c4ebc
1 parent a75571d
commit 19c4ebc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flux/sampling.py b/flux/sampling.py
@@ -177,11 +177,11 @@ def denoise(
             compile_run,
             image_latents,
             mask,
-            noise[i]
+            None if noise is None else noise[i]
         )
         compile_run = False
         output_imgs.append(denoised_img)
-    
+
     return torch.cat(output_imgs), model
 
 def unpack(x: Tensor, height: int, width: int) -> Tensor: