Skip to content

Commit 35d38cb

Browse files
authored
Merge pull request inclusionAI#10 from inclusionAI/update_vllm_readme
Update README.md
2 parents 4260171 + 1922cd7 commit 35d38cb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ tokenizer = AutoTokenizer.from_pretrained("inclusionAI/Ling-lite")
108108

109109
sampling_params = SamplingParams(temperature=0.7, top_p=0.8, repetition_penalty=1.05, max_tokens=512)
110110

111-
llm = LLM(model="inclusionAI/Ling-lite",
111+
llm = LLM(model="inclusionAI/Ling-lite", dtype='bfloat16')
112112
prompt = "Give me a short introduction to large language models."
113113
messages = [
114114
{"role": "system", "content": "You are Ling, an assistant created by inclusionAI"},

0 commit comments

Comments
 (0)