Update train_gpt.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: hardik-bhalekar

train_gpt.py

@@ -425,7 +425,7 @@ def quantize_state_dict_ternary(state_dict: dict[str, Tensor], threshold_scale:
             # all zeros
             scales[name] = torch.tensor(0.0)
             ternary[name] = torch.zeros_like(tt, dtype=torch.int8)
-            stats["ternary_payload_bytes"] += tensor_nbytes(ternary[name])
+            stats["ternary_payload_bytes"] += tensor_nbytes(ternary[name]) + tensor_nbytes(scales[name])
             continue
         thr = threshold_scale * max_abs
         s = max_abs if max_abs > 0 else 1.0