'import AutoConfig' added

README.md

@@ -213,7 +213,7 @@ The reason for writing the code this way is to avoid errors that occur during mu
 ```python
 import math
 import torch
-from transformers import AutoTokenizer, AutoModel
+from transformers import AutoTokenizer, AutoModel, AutoConfig
 
 def split_model(model_name):
     device_map = {}