Yuchan5386 commited on
Commit
6312036
·
verified ·
1 Parent(s): 8a4883f

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -139,7 +139,7 @@ model.load_weights("InteractGPT.weights.h5")
139
  print("모델 가중치 로드 완료!")
140
 
141
  def generate_text_typical(model, prompt, max_len=100, max_gen=98,
142
- temperature=0.85, min_len=20,
143
  repetition_penalty=1.2, typical_p=0.80):
144
 
145
  def typical_filtering(logits, typical_p):
 
139
  print("모델 가중치 로드 완료!")
140
 
141
  def generate_text_typical(model, prompt, max_len=100, max_gen=98,
142
+ temperature=0.50, min_len=20,
143
  repetition_penalty=1.2, typical_p=0.80):
144
 
145
  def typical_filtering(logits, typical_p):