mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 15:02:01 +00:00
lint
This commit is contained in:
parent
347fdbc794
commit
f48712ee29
|
@ -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
|
||||
|
||||
import refiners.fluxion.layers as fl
|
||||
|
||||
assert issubclass(LayerNorm2d, nn.Module)
|
||||
custom_layers = {LayerNorm2d: fl.LayerNorm2d}
|
||||
|
||||
|
|
Loading…
Reference in a new issue