Merge pull request bryandlee#26 from AK391/main

add side by side option to torch hub model
NgoVanDau · Nov 16, 2021 · dd91006 · dd91006
2 parents 6c2de96 + 50b4c15
commit dd91006
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hubconf.py b/hubconf.py
@@ -33,15 +33,15 @@ def generator(pretrained=True, device="cpu", progress=True, check_hash=True):
     return model
 
 
-def face2paint(device="cpu", size=512):
+def face2paint(device="cpu", size=512, side_by_side=False):
     from PIL import Image
     from torchvision.transforms.functional import to_tensor, to_pil_image
 
     def face2paint(
         model: torch.nn.Module,
         img: Image.Image,
         size: int = size,
-        side_by_side: bool = True,
+        side_by_side: bool = side_by_side,
         device: str = device,
     ) -> Image.Image:
         w, h = img.size