2026-04-20 20:58:50.315737: E external/local_xla/xla/stream_executor/cuda/cuda_platform.cc:51] failed call to cuInit: INTERNAL: CUDA error: Failed call to cuInit: UNKNOWN ERROR (303) I0420 20:58:51.441173 136854324661376 max_utils.py:238] Skipping jax distributed system due to skip_jax_distributed_system=True flag. Traceback (most recent call last): File "<frozen runpy>", line 198, in _run_module_as_main File "<frozen runpy>", line 88, in _run_code File "~/maxtext/src/maxtext/trainers/post_train/distillation/train_distill.py", line 777, in <module> app.run(main) File "~/maxtext/maxtext_pt_venv/lib/python3.12/site-packages/absl/app.py", line 367, in run _run_main(main, args) File "~/maxtext/maxtext_pt_venv/lib/python3.12/site-packages/absl/app.py", line 312, in _run_main sys.exit(main(argv)) ^^^^^^^^^^ File "~/maxtext/src/maxtext/trainers/post_train/distillation/train_distill.py", line 752, in main student_config = pyconfig.initialize(argv, **student_overrides) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "~/maxtext/src/maxtext/configs/pyconfig.py", line 294, in initialize pydantic_config = initialize_pydantic(argv, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "~/maxtext/src/maxtext/configs/pyconfig.py", line 343, in initialize_pydantic validate_no_keys_overridden_twice(model_loaded_cfg.keys(), overrides_cfg.keys()) File "~/maxtext/src/maxtext/configs/pyconfig.py", line 99, in validate_no_keys_overridden_twice raise ValueError( ValueError: Keys ['vocab_size'] are overridden by both model config and CLI/kwargs.This is not allowed, unless setting `override_model_config=True`. [DECOUPLED NO-OP] gcs_storage: using stubs.