Commit 0fbe994c authored by Xing Han Lu's avatar Xing Han Lu

Change repo username

parent 6310782e
...@@ -7,7 +7,7 @@ You can load Animegan v2 via `torch.hub`: ...@@ -7,7 +7,7 @@ You can load Animegan v2 via `torch.hub`:
```python ```python
import torch import torch
model = torch.hub.load('xhlulu/animegan2-pytorch', 'generator').eval() model = torch.hub.load('bryandlee/animegan2-pytorch', 'generator').eval()
# convert your image into tensor here # convert your image into tensor here
out = model(img_tensor) out = model(img_tensor)
``` ```
...@@ -15,7 +15,7 @@ out = model(img_tensor) ...@@ -15,7 +15,7 @@ out = model(img_tensor)
You can load with various configs (more details in [the torch docs](https://pytorch.org/docs/stable/hub.html)): You can load with various configs (more details in [the torch docs](https://pytorch.org/docs/stable/hub.html)):
```python ```python
model = torch.hub.load( model = torch.hub.load(
"xhlulu/animegan2-pytorch", "bryandlee/animegan2-pytorch",
"generator", "generator",
pretrained=True, # or give URL to a pretrained model pretrained=True, # or give URL to a pretrained model
device="cuda", # or "cpu" if you don't have a GPU device="cuda", # or "cpu" if you don't have a GPU
...@@ -25,10 +25,10 @@ model = torch.hub.load( ...@@ -25,10 +25,10 @@ model = torch.hub.load(
Currently, the following `pretrained` shorthands are available: Currently, the following `pretrained` shorthands are available:
```python ```python
model = torch.hub.load("xhlulu/animegan2-pytorch", "generator", pretrained="celeba_distill") model = torch.hub.load("bryandlee/animegan2-pytorch", "generator", pretrained="celeba_distill")
model = torch.hub.load("xhlulu/animegan2-pytorch", "generator", pretrained="face_paint_512_v1") model = torch.hub.load("bryandlee/animegan2-pytorch", "generator", pretrained="face_paint_512_v1")
model = torch.hub.load("xhlulu/animegan2-pytorch", "generator", pretrained="face_paint_512_v2") model = torch.hub.load("bryandlee/animegan2-pytorch", "generator", pretrained="face_paint_512_v2")
model = torch.hub.load("xhlulu/animegan2-pytorch", "generator", pretrained="paprika") model = torch.hub.load("bryandlee/animegan2-pytorch", "generator", pretrained="paprika")
``` ```
You can also load the `face2paint` util function. First, install dependencies: You can also load the `face2paint` util function. First, install dependencies:
...@@ -40,7 +40,7 @@ pip install torchvision Pillow numpy ...@@ -40,7 +40,7 @@ pip install torchvision Pillow numpy
Then, import the function using `torch.hub`: Then, import the function using `torch.hub`:
```python ```python
face2paint = torch.hub.load( face2paint = torch.hub.load(
'xhlulu/animegan2-pytorch', 'face2paint', 'bryandlee/animegan2-pytorch', 'face2paint',
size=512, device="cpu" size=512, device="cpu"
) )
......
...@@ -3,7 +3,7 @@ import torch ...@@ -3,7 +3,7 @@ import torch
def generator(pretrained=True, device="cpu", progress=True, check_hash=True): def generator(pretrained=True, device="cpu", progress=True, check_hash=True):
from model import Generator from model import Generator
release_url = "https://github.com/xhlulu/animegan2-pytorch/raw/main/weights" release_url = "https://github.com/bryandlee/animegan2-pytorch/raw/main/weights"
known = { known = {
name: f"{release_url}/{name}.pt" name: f"{release_url}/{name}.pt"
for name in [ for name in [
......
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