Update README.md

README.md

@@ -24,6 +24,7 @@ language:
 
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers.generation import GenerationConfig
 import torch
 torch.manual_seed(1234)
 model_path = 'ChatTruth-7B' # your downloaded model path.
@@ -33,7 +34,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
 # use cuda device
 model = AutoModelForCausalLM.from_pretrained(model_path, device_map="cuda", trust_remote_code=True).eval()
 
-model.generation_config = GenerationConfig.from_pretrained(checkpoint, trust_remote_code=True)
+model.generation_config = GenerationConfig.from_pretrained(model_path, trust_remote_code=True)
 model.generation_config.top_p = 0.01
 
 query = tokenizer.from_list_format([