Fix another import

This commit is contained in:
James Betker 2020-11-14 22:10:45 -07:00
parent 9af049c671
commit 89f56b2091

View File

@ -4,7 +4,7 @@ from math import log2
import torch
import torch.nn as nn
from models.archs.stylegan.stylegan2 import attn_and_ff
from models.archs.stylegan.stylegan2 import attn_and_ff, Flatten
def leaky_relu(p=0.2):