mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-24 15:18:46 +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
|
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}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue