Merge pull request #1 from qsun1/qsun1-patch-1

fix a bug in stylegan2_generator.py
qsun1 · Jul 25, 2023 · af04744 · af04744
2 parents 3f1df30 + 3cc5f96
commit af04744
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmagic/models/editors/stylegan2/stylegan2_generator.py b/mmagic/models/editors/stylegan2/stylegan2_generator.py
@@ -536,7 +536,7 @@ def forward(self,
         img = skip.to(torch.float32)
 
         if self.bgr2rgb:
-            img = torch.flip(img, dims=1)
+            img = torch.flip(img, dims=[1])
 
         if return_latents or return_noise:
             output_dict = dict(