LMFuser 0.0.1__tar.gz → 0.0.2__tar.gz

{lmfuser-0.0.1 → lmfuser-0.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: LMFuser
-Version: 0.0.1
+Version: 0.0.2
 Summary: The LMFuser training framework.
 Project-URL: Homepage, https://github.com/TYTTYTTYT/LMFuser
 Project-URL: Documentation, https://github.com/TYTTYTTYT/LMFuser

{lmfuser-0.0.1 → lmfuser-0.0.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "LMFuser"
-version = "0.0.1"
+version = "0.0.2"
 requires-python = ">= 3.11"
 description = "The LMFuser training framework."
 readme = "README.md"
@@ -40,4 +40,4 @@ Repository = "https://github.com/TYTTYTTYT/LMFuser"
 Changelog = "https://github.com/TYTTYTTYT/LMFuser"
 
 [tool.hatch.build.targets.wheel]
-packages = ["src/lmfuser"]
+packages = ["src/lmfuser"]

{lmfuser-0.0.1 → lmfuser-0.0.2}/src/lmfuser/runners/ddp_runner.py

@@ -432,7 +432,7 @@ class DDPRunner(Runner[DDPRunnerConfig]):
                     if isinstance(v, (float, int)):
                         batch_datas[k].append(float(v))
                     elif isinstance(v, (list)) and len(v) > 0 and isinstance(v[0], (float, int)):
-                        batch_datas[k].extend([float(i) for i in v])
+                        batch_datas[k].extend([float(i) for i in v]) # type: ignore
 
             if self.scaler is not None:
                 self.scaler.scale(loss).backward()
@@ -524,9 +524,10 @@ class DDPRunner(Runner[DDPRunnerConfig]):
         self._prepare_train()
 
         self._pbar_train = tqdm(
-            total=self.config.total_step.value(), 
-            position=0, 
-            dynamic_ncols=True, 
+            total=self.config.total_step.value(),
+            position=0,
+            initial=self.step - 1,
+            dynamic_ncols=True,
             unit='step',
             disable=True if get_global_rank() != 0 else False
         )