amd
/

Zebra-Llama-8B-8MLA-24Mamba-SFT

alignment-handbook

Generated from Trainer

Model card Files Files and versions

Mingyuyang-1 commited on Sep 23

Commit

1cea748

·

verified ·

1 Parent(s): 79202b2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -70,7 +70,7 @@ Once the installation completed, we can try the following code for a quick test
 ```python
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-from hybrid_model.hybrid_model_wrapper import HybridTransformerHybridModelWrapper
 checkpoint = "amd/Zebra-Llama-8B-8MLA-24Mamba-SFT"

 ```python
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from hybrid.hybrid_wrapper import HybridModelWrapper
 checkpoint = "amd/Zebra-Llama-8B-8MLA-24Mamba-SFT"