SubspaceDecoder_mla192-96-192 / modeling_shared_space_decoder.py
ChrisMcCormick's picture
Debugging
ff0637b verified
# Re-export to satisfy "module.ClassName" dynamic import
from .models.shared_space_decoder import SharedSpaceDecoderModel
from .layers.task_heads import SharedSpaceDecoderForCausalLM