Yuchan commited on
Commit
e37c07d
·
verified ·
1 Parent(s): 78ad98c

Update Mo.py

Browse files
Files changed (1) hide show
  1. Mo.py +1 -1
Mo.py CHANGED
@@ -276,7 +276,7 @@ history = model.fit(dataset, epochs=1, verbose=1)
276
  model.save_weights("model.weights.h5")
277
  print("모델 가중치 저장 완료!")
278
 
279
- def generate_text_topp(model, prompt, max_len=150, max_gen=150, p=0.9, temperature=0.8, min_len=20):
280
  model_input = text_to_ids(f"<start> {prompt}")
281
  model_input = model_input[:max_len]
282
  generated = list(model_input)
 
276
  model.save_weights("model.weights.h5")
277
  print("모델 가중치 저장 완료!")
278
 
279
+ def generate_text_topp(model, prompt, max_len=512, max_gen=512, p=0.9, temperature=0.8, min_len=20):
280
  model_input = text_to_ids(f"<start> {prompt}")
281
  model_input = model_input[:max_len]
282
  generated = list(model_input)