Update README.md

README.md

@@ -31,7 +31,7 @@ git clone https://huggingface.co/TerenceLau/sparrow && cd ./sparrow/
 
 then, you can register the `SparrowConfig` and `SparrowModel` to `AutoConfig` and `AutoModelForCausalLM`:
 ```python
-from modelling_sparrow import SparrowModel, SparrowConfig
+from modelling_sparrow import SparrowModelForCausalLM, SparrowConfig
 from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer
 AutoConfig.register("sparrow", SparrowConfig)
 AutoModelForCausalLM.register(SparrowConfig, SparrowModelForCausalLM)