Skip to content

Commit 4c693e2

Browse files
lingvo-botcopybara-github
authored andcommitted
Allow using inference graph exporter in eager mode, since it works as long as we are using graph context.
PiperOrigin-RevId: 477530819
1 parent 50d55da commit 4c693e2

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

lingvo/core/inference_graph_exporter.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -344,8 +344,6 @@ def Export(cls,
344344
Raises:
345345
ValueError: if the model does not support the listed subgraphs.
346346
"""
347-
if py_utils.IsEagerMode():
348-
raise ValueError('InferenceGraph exporter does not work in Eager mode.')
349347
assert issubclass(model_cfg.cls, base_model.BaseModel)
350348
if device_options.dtype_override and device_options.fprop_dtype_override:
351349
raise ValueError(

0 commit comments

Comments
 (0)