SplatAnything / src /models /layers /__init__.py
ritianyu's picture
Publish preview-first SplatAnything demo for ZeroGPU
dc09dcc verified
Raw History Blame Contribute Delete
275 Bytes
# Modified for SplatAnything: inference-only release.
from .mlp import Mlp, MlpFP32
from .patch_embed import PatchEmbed, PatchEmbed_Mlp
from .swiglu_ffn import SwiGLUFFN, SwiGLUFFNFused
from .block import NestedTensorBlock
from .attention import Attention as MemEffAttention