We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 81eca03 commit d87ab8aCopy full SHA for d87ab8a
src/MaxText/utils/ckpt_conversion/to_maxtext.py
@@ -61,7 +61,7 @@
61
import threading
62
from functools import partial
63
from typing import Sequence, List, Any, Callable
64
-from benchmarks.benchmark_utils import str2bool
+from MaxText.inference_utils import str2bool
65
import numpy as np
66
import jax
67
import psutil
0 commit comments