Skip to content

Commit

Permalink
Update hubconf.py
Browse files Browse the repository at this point in the history
  • Loading branch information
AK391 committed Nov 16, 2021
1 parent fab8066 commit a246b39
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions hubconf.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,15 +33,15 @@ def generator(pretrained=True, device="cpu", progress=True, check_hash=True):
return model


def face2paint(device="cpu", size=512, side_by_side=True):
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 = False,
device: str = device,
) -> Image.Image:
w, h = img.size
Expand Down

0 comments on commit a246b39

Please sign in to comment.