cant remove add noise function in img to img

[walt798]

i am trying to use streamdiffusion to create an ai video where each frame is one step of the ai by using the stream() function with only one inference step but i am getting very blurry outputs. i dug into the code and found that every time img2img is used it adds some noise to the input image in the add_noise function. you cannot disable this. i tried changing this in pipeline.py:

def add_noise(
        self,
        original_samples: torch.Tensor,
        noise: torch.Tensor,
        t_index: int,
    ) -> torch.Tensor:
        noisy_samples = (
            self.alpha_prod_t_sqrt[t_index] * original_samples
            + self.beta_prod_t_sqrt[t_index] * noise 
        )
        
        return noisy_samples

to this:

def add_noise(
        self,
        original_samples: torch.Tensor,
        noise: torch.Tensor,
        t_index: int,
    ) -> torch.Tensor:
        noisy_samples = (
            self.alpha_prod_t_sqrt[t_index] * original_samples
            
        )

        
        return noisy_samples

as you can see i removed where the noise is added but due to my lack of knowledge this just makes it so the image turns into a black blob.

any ideas how to disable noise in image to image streamdiffusion by editing the code?

Note:this isnt really a big issue but there really should be a way to disable this.