Unverified Commit 50b4c151 authored by AK391's avatar AK391 Committed by GitHub

Update hubconf.py

parent a246b391
...@@ -41,7 +41,7 @@ def face2paint(device="cpu", size=512, side_by_side=False): ...@@ -41,7 +41,7 @@ def face2paint(device="cpu", size=512, side_by_side=False):
model: torch.nn.Module, model: torch.nn.Module,
img: Image.Image, img: Image.Image,
size: int = size, size: int = size,
side_by_side: bool = False, side_by_side: bool = side_by_side,
device: str = device, device: str = device,
) -> Image.Image: ) -> Image.Image:
w, h = img.size w, h = img.size
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment