Fix another import
This commit is contained in:
parent
9af049c671
commit
89f56b2091
|
@ -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):
|
||||
|
|
Loading…
Reference in New Issue
Block a user