Yuchan commited on
Commit
094d892
Β·
verified Β·
1 Parent(s): 6c265e3

Update AlphaS2S.py

Browse files
Files changed (1) hide show
  1. AlphaS2S.py +2 -2
AlphaS2S.py CHANGED
@@ -296,7 +296,7 @@ chat_model.save_weights("chat_model.weights.h5")
296
  print("\nβœ… λͺ¨λΈ κ°€μ€‘μΉ˜ μ €μž₯ μ™„λ£Œ!")
297
 
298
 
299
- def generate_translation_beam(model, input_text, max_len=220, beam_width=5):
300
  # Encoder input
301
  enc_ids = text_to_ids(input_text)
302
  enc_ids = enc_ids[-max_len:]
@@ -342,5 +342,5 @@ def generate_translation_beam(model, input_text, max_len=220, beam_width=5):
342
 
343
  # μ‚¬μš© μ˜ˆμ‹œ
344
  src_text = "μ•ˆλ…•ν•˜μ„Έμš”! 였늘 λ‚ μ”¨λŠ” μ–΄λ•Œμš”?"
345
- translation = generate_translation_beam(chat_model, src_text, max_len=220, beam_width=5)
346
  print("λ²ˆμ—­ κ²°κ³Ό:", translation)
 
296
  print("\nβœ… λͺ¨λΈ κ°€μ€‘μΉ˜ μ €μž₯ μ™„λ£Œ!")
297
 
298
 
299
+ def generate_translation_beam(model, input_text, max_len=128, beam_width=5):
300
  # Encoder input
301
  enc_ids = text_to_ids(input_text)
302
  enc_ids = enc_ids[-max_len:]
 
342
 
343
  # μ‚¬μš© μ˜ˆμ‹œ
344
  src_text = "μ•ˆλ…•ν•˜μ„Έμš”! 였늘 λ‚ μ”¨λŠ” μ–΄λ•Œμš”?"
345
+ translation = generate_translation_beam(chat_model, src_text, max_len=128, beam_width=5)
346
  print("λ²ˆμ—­ κ²°κ³Ό:", translation)