Compatibility with standard PEFT / QLoRA pipelines (Unsloth, TRL) for the custom Mamba-2/MoE architecture?

#1
by Mightysweet - opened

Hey Soofi team,

Thank you for releasing the preview models! The custom hybrid Mamba-2/MoE architecture looks very promising.

Since the model relies on custom modeling code (trust_remote_code=True), I have a question regarding fine-tuning:
Is this model out-of-the-box compatible with standard PEFT / QLoRA pipelines like Unsloth or the Hugging Face TRL library? Or does the unique combination of Mamba-2 SSM layers and MoE routing require specialized training scripts and custom implementations to apply LoRA/QLoRA successfully?

If there are any recommended workflows, existing scripts, or planned support for tools like Unsloth, I would greatly appreciate any pointers.

Thanks in advance for your time and help!

Sebastian

Sign up or log in to comment