Update app.py

app.py

@@ -689,12 +689,13 @@ class AccDiffusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoade
         seed: Optional[int] = None,
         c : Optional[float] = 0.3,
     ):
-
         """
         Stable Diffusion XL 기반 AccDiffusion 파이프라인을 통해 이미지를 생성하는 함수입니다.
 
         이 함수는 주어진 프롬프트를 인코딩하고, 디노이징 및 progressive upscaling 과정을 거쳐 최종 이미지를 생성합니다.
         자세한 사용법은 문서를 참고하세요.
+
+        Examples:
         """
     
         if debug:
@@ -1093,8 +1094,8 @@ class AccDiffusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoade
                                 count_local[:, :, h_start:h_end, w_start:w_end] += 1
 
                         if random_jitter:
-                            value_local = value_local[: ,:, jitter_range: jitter_range + current_height // self.vae_scale_factor, jitter_range: jitter_range + current_width // self.vae_scale_factor]
-                            count_local = count_local[: ,:, jitter_range: jitter_range + current_height // self.vae_scale_factor, jitter_range: jitter_range + current_width // self.vae_scale_factor]
+                            value_local = value_local[:, :, jitter_range:jitter_range + current_height // self.vae_scale_factor, jitter_range:jitter_range + current_width // self.vae_scale_factor]
+                            count_local = count_local[:, :, jitter_range:jitter_range + current_height // self.vae_scale_factor, jitter_range:jitter_range + current_width // self.vae_scale_factor]
 
                         noise_index = i + 1 if i != (len(timesteps) - 1) else i
 
@@ -1145,12 +1146,12 @@ class AccDiffusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoade
 
                             if shuffle:
                                 shape = latents_for_view.shape
+                                # 수정: range(...)의 괄호를 추가합니다.
                                 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])
+                                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)
-                                original_index = original_index.permute(3,0,1,2).to(device)
+                                shuffle_index = shuffle_index.permute(3, 0, 1, 2).to(device)
+                                original_index = original_index.permute(3, 0, 1, 2).to(device)
                                 latents_for_view_gaussian = latents_for_view_gaussian.gather(0, shuffle_index)
                         
                             vb_size = latents_for_view.size(0)
@@ -1195,7 +1196,7 @@ class AccDiffusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoade
                                 value_global[:, :, h::current_scale_num, w::current_scale_num] += latents_view_denoised
                                 count_global[:, :, h::current_scale_num, w::current_scale_num] += 1
                                 
-                        value_global = value_global[: ,:, h_pad:, w_pad:]
+                        value_global = value_global[:, :, h_pad:, w_pad:]
 
                         if use_multidiffusion:
                             c2 = cosine_factor ** cosine_scale_2