This commit is contained in:
limiteinductive 2024-04-16 13:56:56 +00:00 committed by Benjamin Trom
parent 347fdbc794
commit f48712ee29

View file

@ -258,8 +258,6 @@ def test_mask_decoder(facebook_sam_h: FacebookSAM, sam_h: SegmentAnythingH) -> N
from segment_anything.modeling.common import LayerNorm2d # type: ignore from segment_anything.modeling.common import LayerNorm2d # type: ignore
import refiners.fluxion.layers as fl
assert issubclass(LayerNorm2d, nn.Module) assert issubclass(LayerNorm2d, nn.Module)
custom_layers = {LayerNorm2d: fl.LayerNorm2d} custom_layers = {LayerNorm2d: fl.LayerNorm2d}