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

Hi @Mightysweet ,

Just curious how you got access to the model. Are you under NDA or how was the process as most people seem to be blocked out?

Best
Tim

Sign up or log in to comment