Yuchan commited on
Commit
703f5ae
·
verified ·
1 Parent(s): 85d30e7

Update Model.py

Browse files
Files changed (1) hide show
  1. Model.py +1 -1
Model.py CHANGED
@@ -366,7 +366,7 @@ history = model.fit(
366
  model.save_weights("Cobra.weights.h5")
367
  print("모델 가중치 저장 완료!")
368
 
369
- def generate_text_topp(model, prompt, max_len=100, max_gen=98, p=0.9, temperature=0.8, min_len=20):
370
  model_input = text_to_ids(f"<start> {prompt} <sep>")
371
  model_input = model_input[:max_len]
372
  generated = list(model_input)
 
366
  model.save_weights("Cobra.weights.h5")
367
  print("모델 가중치 저장 완료!")
368
 
369
+ def generate_text_topp(model, prompt, max_len=512, max_gen=512, p=0.9, temperature=0.8, min_len=20):
370
  model_input = text_to_ids(f"<start> {prompt} <sep>")
371
  model_input = model_input[:max_len]
372
  generated = list(model_input)