Yuchan5386 commited on
Commit
11d37d9
·
verified ·
1 Parent(s): 9ab8176

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +2 -2
api.py CHANGED
@@ -139,8 +139,8 @@ model.load_weights("InteractGPT.weights.h5")
139
  print("모델 가중치 로드 완료!")
140
 
141
  def generate_text_top_kp(model, prompt, max_len=100, max_gen=98,
142
- temperature=1.0, min_len=20,
143
- repetition_penalty=1.1, top_k=40, top_p=0.9):
144
  model_input = text_to_ids(f"<start> {prompt} <sep>")
145
  model_input = model_input[:max_len]
146
  generated = list(model_input)
 
139
  print("모델 가중치 로드 완료!")
140
 
141
  def generate_text_top_kp(model, prompt, max_len=100, max_gen=98,
142
+ temperature=0.7, min_len=20,
143
+ repetition_penalty=1.1, top_k=50, top_p=0.9):
144
  model_input = text_to_ids(f"<start> {prompt} <sep>")
145
  model_input = model_input[:max_len]
146
  generated = list(model_input)