Join MaskNet to guide generator

This commit is contained in:
henryruhs
2025-03-16 12:24:08 +01:00
parent ad675ae633
commit c45fcbba84
+1 -1
View File
@@ -4,8 +4,8 @@ from typing import Tuple
from torch import Tensor, nn
from ..networks.aad import AAD
from ..networks.unet import UNet
from ..networks.masknet import MaskNet
from ..networks.unet import UNet
from ..types import Embedding, Feature, Mask