germanjke commited on
Commit
f88b5bf
1 Parent(s): dd86e71

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -127,7 +127,9 @@ from vllm import LLM, SamplingParams
127
  model_name = "t-tech/T-pro-it-1.0"
128
  tokenizer = AutoTokenizer.from_pretrained(model_name)
129
  llm = LLM(model=model_name, max_model_len=8192)
130
- sampling_params = SamplingParams(temperature=0.3, max_tokens=8192)
 
 
131
 
132
  prompt = "Напиши стих про машинное обучение"
133
  messages = [
 
127
  model_name = "t-tech/T-pro-it-1.0"
128
  tokenizer = AutoTokenizer.from_pretrained(model_name)
129
  llm = LLM(model=model_name, max_model_len=8192)
130
+ sampling_params = SamplingParams(temperature=0.7,
131
+ repetition_penalty=1.05,
132
+ top_p=0.8, top_k=70)
133
 
134
  prompt = "Напиши стих про машинное обучение"
135
  messages = [