OpenLab-NLP
/

model-prototype

Model card Files Files and versions

Yuchan commited on Nov 23

Commit

0b7b202

·

verified ·

1 Parent(s): 26b3b8c

Update Inference.py

Files changed (1) hide show

Inference.py +1 -1

Inference.py CHANGED Viewed

@@ -187,7 +187,7 @@ model = ReLM(
     n_layers=1
 )
-chat_model.load_weights('/kaggle/working/model.weights.h5')
 print("모델 가중치 로드 완료!")
 # =======================
 # 6) 추론 함수 (기존 코드 유지)

     n_layers=1
 )
+model.load_weights('/kaggle/working/model.weights.h5')
 print("모델 가중치 로드 완료!")
 # =======================
 # 6) 추론 함수 (기존 코드 유지)