Update strength image to image gen

5c0cd1839bdddd549a0a3f51e97064f7cc58ff8e · 2024-12-16 18:40:36 -0500 · Pranav Veldurthi

Files touched

M exo/inference/mlx/models/StableDiffusionPipeline.py

Diff

commit 5c0cd1839bdddd549a0a3f51e97064f7cc58ff8e
Author: Pranav Veldurthi <veldurthipranav@gmail.com>
Date:   Mon Dec 16 18:40:36 2024 -0500

    Update strength image to image gen
---
 exo/inference/mlx/models/StableDiffusionPipeline.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exo/inference/mlx/models/StableDiffusionPipeline.py b/exo/inference/mlx/models/StableDiffusionPipeline.py
index 1f3bd1ca..f8e7c054 100644
--- a/exo/inference/mlx/models/StableDiffusionPipeline.py
+++ b/exo/inference/mlx/models/StableDiffusionPipeline.py
@@ -181,7 +181,7 @@ class Model(nn.Module):
         else:
             self.decoder = nn.Identity()            
 
-    def __call__(self,x, step= 0, cfg_weight: float = 7.5,total_steps=50,conditioning=None,mask=None,residual=None,x_t_prev=None,is_finished=False,is_step_finished=False, image=None, strength=0.7, start_step=None):
+    def __call__(self,x, step= 0, cfg_weight: float = 7.5,total_steps=50,conditioning=None,mask=None,residual=None,x_t_prev=None,is_finished=False,is_step_finished=False, image=None, strength=0.65, start_step=None):
         t, t_prev = self.sampler.current_timestep(step=step, total_steps=total_steps, start_time=start_step)
         is_finished = False
         is_step_finished = False

← e2474c3f fail if we never get the desired node count · back to Exo · optimise grpc parameters 58f0a0f5 →