Update

yangxy · Mar 4, 2022 · bad3749 · bad3749
1 parent 0634e38
commit bad3749
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/face_model/gpen_model.py b/face_model/gpen_model.py
@@ -552,8 +552,7 @@ def forward(
             return image, latent
 
         else:
-            #return image, None
-            return image
+            return image, None
 
 class ConvLayer(nn.Sequential):
     def __init__(