Yuchan5386 commited on
Commit
a45d3f4
·
verified ·
1 Parent(s): 19b65f7

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -141,7 +141,7 @@ model = InteractGPT(vocab_size=vocab_size, seq_len=max_len, d_model=256, d_ff=10
141
 
142
  dummy_input = tf.zeros((1, max_len), dtype=tf.int32) # 배치1, 시퀀스길이 max_len
143
  _ = model(dummy_input) # 모델이 빌드됨
144
- model.load_weights("InteractGPT.weights.h5")
145
  print("모델 가중치 로드 완료!")
146
 
147
 
 
141
 
142
  dummy_input = tf.zeros((1, max_len), dtype=tf.int32) # 배치1, 시퀀스길이 max_len
143
  _ = model(dummy_input) # 모델이 빌드됨
144
+ model.load_weights("Flexi.weights.h5")
145
  print("모델 가중치 로드 완료!")
146
 
147