Yuchan5386 commited on
Commit
feb045d
·
verified ·
1 Parent(s): 972db66

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -149,7 +149,7 @@ print("모델 가중치 로드 완료!")
149
 
150
 
151
  def generate_text_sample(model, prompt, max_len=100, max_gen=98,
152
- temperature=0.8, top_k=55, top_p=0.95, min_len=12):
153
  model_input = text_to_ids(f"{prompt}")
154
  model_input = model_input[:max_len]
155
  generated = list(model_input)
 
149
 
150
 
151
  def generate_text_sample(model, prompt, max_len=100, max_gen=98,
152
+ temperature=0.85, top_k=65, top_p=0.9725, min_len=12):
153
  model_input = text_to_ids(f"{prompt}")
154
  model_input = model_input[:max_len]
155
  generated = list(model_input)