提交 30d6a746 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Suggest tfp-nighly

上级 3b722cec
......@@ -44,7 +44,7 @@ def try_import_tfp_jax_op(op: ScalarOp, jax_op_name: str | None = None) -> Calla
except ModuleNotFoundError:
raise NotImplementedError(
f"No JAX implementation for Op {op.name}. "
"Implementation is available if TensorFlow Probability is installed"
"Implementation is available if tfp-nightly is installed"
)
if jax_op_name is None:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论