From 4b681169777a6ef6382953e1f255483a8d2c39b7 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sun, 15 Nov 2020 16:15:42 -0700 Subject: [PATCH] import fix --- codes/models/archs/stylegan/stylegan2.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codes/models/archs/stylegan/stylegan2.py b/codes/models/archs/stylegan/stylegan2.py index f29475ce..16735af3 100644 --- a/codes/models/archs/stylegan/stylegan2.py +++ b/codes/models/archs/stylegan/stylegan2.py @@ -8,6 +8,7 @@ from random import random import torch import torch.nn.functional as F import models.steps.losses as L +import numpy as np from kornia.filters import filter2D from linear_attention_transformer import ImageLinearAttention