From 3d773c0a486dd197e543b3761c223934cb4cd0bf Mon Sep 17 00:00:00 2001
From: Mr-Neutr0n <64578610+Mr-Neutr0n@users.noreply.github.com>
Date: Sat, 7 Feb 2026 02:08:54 +0530
Subject: [PATCH] Add missing math and sys imports in llava_trainer.py

The _inner_training_loop method uses math.ceil() and sys.maxsize but
neither module was imported, causing NameError during training.

Fixes #504
---
 llava/train/llava_trainer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/llava/train/llava_trainer.py b/llava/train/llava_trainer.py
index f8e0132bb..485292ad0 100755
--- a/llava/train/llava_trainer.py
+++ b/llava/train/llava_trainer.py
@@ -1,4 +1,6 @@
+import math
 import os
+import sys
 import torch
 import torch.nn as nn
 import datetime