We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8671459 commit 3a044c5Copy full SHA for 3a044c5
1 file changed
src/instructlab/training/utils.py
@@ -787,9 +787,7 @@ def _get_state_dict_patched(model, unwrap=False):
787
if accelerator.is_main_process:
788
from instructlab.training.vlm_utils import requantize_fp8_state_dict
789
790
- log_rank_0(
791
- "Re-quantizing FP8 parameters for checkpoint compatibility"
792
- )
+ log_rank_0("Re-quantizing FP8 parameters for checkpoint compatibility")
793
model_state = model.module.state_dict()
794
model_state = requantize_fp8_state_dict(
795
model_state, model.module.config._fp8_scales
0 commit comments