Update README.md

README.md

@@ -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
+from hybrid.hybrid_wrapper import HybridModelWrapper
 
 checkpoint = "amd/Zebra-Llama-3B-6MLA-22Mamba-DPO"