Update AlphaS2S.py

AlphaS2S.py

@@ -13,7 +13,7 @@ tf.get_logger().setLevel("ERROR")
 SEED = 42
 tf.random.set_seed(SEED)
 np.random.seed(SEED)
-max_len = 200 # 기존 코드에서 200으로 설정됨
+max_len = 150 # 기존 코드에서 200으로 설정됨
 batch_size = 128
 
 # TPU 초기화 (기존 코드와 동일)
@@ -419,7 +419,7 @@ print("\n✅ 모델 가중치 저장 완료!")
 # 6) 추론 함수 (기존 코드 유지)
 # =======================
 
-def generate_text_topp(model, prompt, max_len=200, max_gen=100, p=0.9, temperature=0.8, min_len=20):
+def generate_text_topp(model, prompt, max_len=150, max_gen=100, p=0.9, temperature=0.8, min_len=20):
     # 인코더 입력은 <start> Prompt <sep> 만 사용
     model_input = text_to_ids(f"<start> {prompt} <sep>")
     model_input = model_input[:max_len]