Update README.md

README.md

@@ -67,7 +67,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
+from hybrid.hybrid_wrapper import HybridModelWrapper
 
 checkpoint = "amd/Zebra-Llama-1B-8MLA-8Mamba-SFT"