Update model.py

model.py

@@ -756,6 +756,7 @@ class StreamMultiDiffusion(nn.Module):
 
         if isinstance(masks, Image.Image):
             masks = [masks]
+        p = self.num_layers
         n = len(masks) if masks is not None else 0
 
         # Modificiation.