Update pixelflow/pipeline_pixelflow.py

pixelflow/pipeline_pixelflow.py

@@ -132,7 +132,7 @@ class PixelFlowPipeline:
 
         return prompt_embeds, prompt_attention_mask
 
-    def sample_block_noise(self, bs, ch, height, width, eps=1e-6)):
+    def sample_block_noise(self, bs, ch, height, width, eps=1e-6):
         gamma = self.scheduler.gamma
         dist = torch.distributions.multivariate_normal.MultivariateNormal(torch.zeros(4), torch.eye(4) * (1 - gamma) + torch.ones(4, 4) * gamma + eps * torch.eye(4))
         block_number = bs * ch * (height // 2) * (width // 2)