Update app.py

app.py

@@ -1151,7 +1151,8 @@ class AccDiffusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoade
 
                             if shuffle:
                                 shape = latents_for_view.shape
-                                shuffle_index = torch.stack([torch.randperm(shape[0]) for _ in range(latents_for_view.reshape(-1).shape[0]//shape[0]])
+                                shuffle_index = torch.stack([torch.randperm(shape[0]) for _ in range(latents_for_view.reshape(-1).shape[0]//shape[0])])
+
                                 shuffle_index = shuffle_index.view(shape[1],shape[2],shape[3],shape[0])
                                 original_index = torch.zeros_like(shuffle_index).scatter_(3, shuffle_index, torch.arange(shape[0]).repeat(shape[1], shape[2], shape[3], 1))
                                 shuffle_index = shuffle_index.permute(3,0,1,2).to(device)