Update README.md
Browse files
README.md
CHANGED
@@ -66,7 +66,7 @@ max_model_len, tp_size = 4096, 1
|
|
66 |
model_name = "th-nuernberg/DeepHermes-3-Mistral-24B-Preview-FP8-Dynamic"
|
67 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
68 |
llm = LLM(model=model_name, tensor_parallel_size=tp_size, max_model_len=max_model_len, trust_remote_code=True)
|
69 |
-
sampling_params = SamplingParams(temperature=0.
|
70 |
|
71 |
messages_list = [
|
72 |
[{"role": "user", "content": "Who are you? Please respond in pirate speak!"}],
|
|
|
66 |
model_name = "th-nuernberg/DeepHermes-3-Mistral-24B-Preview-FP8-Dynamic"
|
67 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
68 |
llm = LLM(model=model_name, tensor_parallel_size=tp_size, max_model_len=max_model_len, trust_remote_code=True)
|
69 |
+
sampling_params = SamplingParams(temperature=0.15, max_tokens=256, stop_token_ids=[tokenizer.eos_token_id])
|
70 |
|
71 |
messages_list = [
|
72 |
[{"role": "user", "content": "Who are you? Please respond in pirate speak!"}],
|