Fix another import
This commit is contained in:
parent
9af049c671
commit
89f56b2091
|
@ -4,7 +4,7 @@ from math import log2
|
||||||
import torch
|
import torch
|
||||||
import torch.nn as nn
|
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):
|
def leaky_relu(p=0.2):
|
||||||
|
|
Loading…
Reference in New Issue
Block a user