pushes

app.py

@@ -3,9 +3,10 @@ os.system("pip install gdown lpips gradio")
 os.system("gdown --id 1HKmjg6iXsWr4aFPuU0gBXPGR83wqMzq7 -O align.dat")
 os.system("wget https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada-pytorch/pretrained/ffhq.pkl")
 os.system("gdown https://github.com/ninja-build/ninja/releases/download/v1.10.2/ninja-linux.zip")
-os.system("unzip -d /usr/local/bin/")
+os.system("unzip ninja-linux.zip -d /usr/local/bin/")
 os.system("sudo update-alternatives --install /usr/bin/ninja ninja /usr/local/bin/ninja 1 --force")
 os.mkdir("embeddings/")
+os.mkdir("images/")
 
 
 import gradio as gr

criteria/localitly_regulizer.py

@@ -1,6 +1,5 @@
 import torch
 import numpy as np
-import wandb
 from criteria import l2_loss
 from configs import hyperparameters
 from configs import global_config

edit.py

@@ -1,4 +1,3 @@
-import wandb
 import click
 import os
 import sys

makedirs.py

@@ -1,4 +1,3 @@
-import wandb
 import click
 import os
 import sys

scripts/latent_creators/base_latent_creator.py

@@ -8,7 +8,7 @@ from random import choice
 from string import ascii_uppercase
 import torch
 from torch.utils.data import DataLoader
-import wandb
+
 from configs import global_config, paths_config
 from tqdm import tqdm
 

scripts/run_pti.py

@@ -4,7 +4,7 @@ from torch.utils.data import DataLoader
 from torchvision.transforms import transforms
 import os
 from configs import global_config, paths_config
-import wandb
+
 
 from training.coaches.multi_id_coach import MultiIDCoach
 from training.coaches.single_id_coach import SingleIDCoach

training/coaches/base_coach.py

@@ -2,7 +2,7 @@ import abc
 import os
 import pickle
 from argparse import Namespace
-import wandb
+
 import os.path
 from criteria.localitly_regulizer import Space_Regulizer
 import torch

training/projectors/w_plus_projector.py

@@ -9,7 +9,6 @@
 """Project given image to the latent space of pretrained network pickle."""
 
 import copy
-import wandb
 import numpy as np
 import torch
 import torch.nn.functional as F

training/projectors/w_projector.py

@@ -9,7 +9,7 @@
 """Project given image to the latent space of pretrained network pickle."""
 
 import copy
-import wandb
+
 import numpy as np
 import torch
 import torch.nn.functional as F

utils/log_utils.py

@@ -1,6 +1,5 @@
 import numpy as np
 from PIL import Image
-import wandb
 from configs import global_config
 import torch
 import matplotlib.pyplot as plt