[Model] use AutoWeightsLoader in model load_weights (#15770)

Signed-off-by: rongfu.leng <rongfu.leng@daocloud.io>
2025-04-02 22:47:31 +08:00
parent 550b2801ad
commit e86c414d6a
4 changed files with 189 additions and 165 deletions
--- a/docs/source/models/supported_models.md
+++ b/docs/source/models/supported_models.md
@@ -218,6 +218,11 @@ See [this page](#generative-models) for more information on how to use generativ
  * `baichuan-inc/Baichuan2-13B-Chat`, `baichuan-inc/Baichuan-7B`, etc.
  * ✅︎
  * ✅︎
+- * `BambaForCausalLM`
+  * Bamba
+  * `ibm-ai-platform/Bamba-9B-fp8`, `ibm-ai-platform/Bamba-9B`
+  *
+  *
 - * `BloomForCausalLM`
  * BLOOM, BLOOMZ, BLOOMChat
  * `bigscience/bloom`, `bigscience/bloomz`, etc.