1580158
/
stable-diffusion-webui-pixelization
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui-pixelization.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Download models with gdown

pull/17/head
missionfloyd 2023-04-02 17:46:31 -06:00
parent e37251c595
commit 7338e1ea3f
3 changed files with 7 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -16,8 +16,6 @@ Download all three models from the table and place them into the `checkpoints` d
| https://drive.google.com/file/d/17f2rKnZOpnO9ATwRXgqLz5u5AZsyDvq_/view?usp=sharing | alias_net.pth | | https://drive.google.com/file/d/17f2rKnZOpnO9ATwRXgqLz5u5AZsyDvq_/view?usp=sharing | alias_net.pth |
| https://drive.google.com/file/d/1i_8xL3stbLWNF4kdQJ50ZhnRFhSDh3Az/view?usp=sharing | 160_net_G_A.pth | | https://drive.google.com/file/d/1i_8xL3stbLWNF4kdQJ50ZhnRFhSDh3Az/view?usp=sharing | 160_net_G_A.pth |
License of Pixelization seems to prevent me from reuploading models anywhere and google drive makes it impossible to download them automatically.
# Credits # Credits
* Original repo: https://github.com/WuZongWei6/Pixelization * Original repo: https://github.com/WuZongWei6/Pixelization

3
install.py
View File

@ -10,3 +10,6 @@ try:
os.remove(os.path.join(path, "pixelization", "models", "__init__.py")) os.remove(os.path.join(path, "pixelization", "models", "__init__.py"))
except OSError as e: except OSError as e:
pass pass
if not launch.is_installed("gdown"):
launch.run_pip("install gdown", "requirements for stable-diffusion-webui-pixelization")

14
scripts/postprocessing_pixelization.py
View File

@ -12,6 +12,7 @@ import numpy as np
from pixelization.models.networks import define_G from pixelization.models.networks import define_G
import pixelization.models.c2pGen import pixelization.models.c2pGen
import gdown
pixelize_code = [ pixelize_code = [
233356.8125, -27387.5918, -32866.8008, 126575.0312, -181590.0156, 233356.8125, -27387.5918, -32866.8008, 126575.0312, -181590.0156,
@ -104,21 +105,14 @@ class Model(torch.nn.Module):
def load(self): def load(self):
os.makedirs(path_checkpoints, exist_ok=True) os.makedirs(path_checkpoints, exist_ok=True)
missing = False
if not os.path.exists(path_pixelart_vgg19): if not os.path.exists(path_pixelart_vgg19):
print(f"Missing {path_pixelart_vgg19} - download it from https://drive.google.com/uc?id=1VRYKQOsNlE1w1LXje3yTRU5THN2MGdMM") gdown.download("https://drive.google.com/uc?id=1VRYKQOsNlE1w1LXje3yTRU5THN2MGdMM", path_pixelart_vgg19)
missing = True
if not os.path.exists(path_160_net_G_A): if not os.path.exists(path_160_net_G_A):
print(f"Missing {path_160_net_G_A} - download it from https://drive.google.com/uc?id=1i_8xL3stbLWNF4kdQJ50ZhnRFhSDh3Az") gdown.download("https://drive.google.com/uc?id=1i_8xL3stbLWNF4kdQJ50ZhnRFhSDh3Az", path_160_net_G_A)
missing = True
if not os.path.exists(path_alias_net): if not os.path.exists(path_alias_net):
print(f"Missing {path_alias_net} - download it from https://drive.google.com/uc?id=17f2rKnZOpnO9ATwRXgqLz5u5AZsyDvq_") gdown.download("https://drive.google.com/uc?id=17f2rKnZOpnO9ATwRXgqLz5u5AZsyDvq_", path_alias_net)
missing = True
assert not missing, 'Missing checkpoints for pixelization - see console for doqwnload links.'
with torch.no_grad(): with torch.no_grad():
self.G_A_net = define_G(3, 3, 64, "c2pGen", "instance", False, "normal", 0.02, [0]) self.G_A_net = define_G(3, 3, 64, "c2pGen", "instance", False, "normal", 0.02, [0])