Update README.md
Browse files
README.md
CHANGED
@@ -50,7 +50,7 @@ from transformers import AutoTokenizer
|
|
50 |
model_id = "bprice9/Palmyra-Med-70B-FP8"
|
51 |
number_gpus = 2
|
52 |
|
53 |
-
sampling_params = SamplingParams(temperature=0.
|
54 |
|
55 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
56 |
|
|
|
50 |
model_id = "bprice9/Palmyra-Med-70B-FP8"
|
51 |
number_gpus = 2
|
52 |
|
53 |
+
sampling_params = SamplingParams(temperature=0.0, top_p=0.9, max_tokens=512, stop_token_ids=[128001, 128009])
|
54 |
|
55 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
56 |
|