Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -203,7 +203,7 @@ class InferenceModel(object):
|
|
203 |
|
204 |
def predict_tokens(self, batch, seed=0):
|
205 |
"""从预处理的数据集批次中预测 tokens。"""
|
206 |
-
print(f"[{current_time()}] 运行:从预处理数据集中预测音符序列")
|
207 |
prediction, _ = self._predict_fn(self._train_state.params, batch, jax.random.PRNGKey(seed))
|
208 |
return self.vocabulary.decode_tf(prediction).numpy()
|
209 |
|
|
|
203 |
|
204 |
def predict_tokens(self, batch, seed=0):
|
205 |
"""从预处理的数据集批次中预测 tokens。"""
|
206 |
+
print(f"[{current_time()}] 运行:从预处理数据集中预测音符序列 (种子:{seed})")
|
207 |
prediction, _ = self._predict_fn(self._train_state.params, batch, jax.random.PRNGKey(seed))
|
208 |
return self.vocabulary.decode_tf(prediction).numpy()
|
209 |
|