Update Mo.py

Mo.py

@@ -245,7 +245,6 @@ def masked_perplexity(y_true, y_pred, eps=0.1):
     mean_loss = tf.reduce_sum(per_tok) / (tf.reduce_sum(mask) + 1e-8)
     return tf.exp(mean_loss)
 
-
 # =======================
 # 모델 생성 & 컴파일
 # =======================
@@ -261,10 +260,8 @@ with strategy.scope():
     # 학습
     history = model.fit(dist_dataset, epochs=1, verbose=1)
 
-
-# 가중치 저장
-model.save_weights("model.weights.h5")
-print("모델 가중치 저장 완료!")
+model.save_weights("tf_model.weights.h5")
+print("✅ 모델 가중치 저장 완료!")
 
 def generate_text_topp(model, prompt, max_len=512, max_gen=512, p=0.9, temperature=0.8, min_len=20):
     model_input = text_to_ids(f"<start> {prompt}")