# Re-export to satisfy "module.ClassName" dynamic import
from .models.shared_space_decoder import SharedSpaceDecoderModel
from .layers.task_heads import SharedSpaceDecoderForCausalLM