Skip to content

Commit 5b094a8

Browse files
[Inference]Fix example in readme (#6178)
1 parent ee81366 commit 5b094a8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

colossalai/inference/README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ engine = InferenceEngine(model, tokenizer, inference_config, verbose=True)
6262

6363
# Step 4: try inference
6464
prompts = ['Who is the best player in the history of NBA?']
65-
response = engine.generate(prompts)
65+
response = engine.generate(prompts=prompts)
6666
pprint(response)
6767
```
6868

0 commit comments

Comments
 (0)