add side by side option

fab8066a · AK391 · GitHub · 6c2de966 · fab8066a
Unverified Commit fab8066a authored Nov 16, 2021 by AK391 Committed by GitHub Nov 16, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

hubconf.py hubconf.py +1 -1

No files found.
--- a/hubconf.py
+++ b/hubconf.py
@@ -33,7 +33,7 @@ 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=True):
    from PIL import Image
    from torchvision.transforms.functional import to_tensor, to_pil_image