gptmed 0.4.0__tar.gz → 0.4.1__tar.gz

{gptmed-0.4.0/gptmed.egg-info → gptmed-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gptmed
-Version: 0.4.0
+Version: 0.4.1
 Summary: A lightweight GPT-based language model framework for training custom question-answering models on any domain
 Author-email: Sanjog Sigdel <sigdelsanjog@gmail.com>
 Maintainer-email: Sanjog Sigdel <sigdelsanjog@gmail.com>

{gptmed-0.4.0 → gptmed-0.4.1}/gptmed/api.py

@@ -203,6 +203,7 @@ def train_from_config(
         log_dir=args['log_dir'],
         device=actual_device,  # Use actual device from DeviceManager
         seed=args['seed'],
+        resume_from=args.get('resume_from'),  # Pass resume checkpoint path
     )
     
     # Create optimizer

{gptmed-0.4.0 → gptmed-0.4.1}/gptmed/configs/train_config.py

@@ -37,6 +37,7 @@ COMMON FAILURE MODES:
 
 from dataclasses import dataclass
 from pathlib import Path
+from typing import Optional
 import json
 
 
@@ -91,6 +92,9 @@ class TrainingConfig:
     # Reproducibility
     seed: int = 42
 
+    # Resume training
+    resume_from: Optional[str] = None  # Path to checkpoint to resume from
+
     def to_dict(self) -> dict:
         """Convert to dictionary."""
         return {
@@ -117,6 +121,7 @@ class TrainingConfig:
             "device": self.device,
             "use_amp": self.use_amp,
             "seed": self.seed,
+            "resume_from": self.resume_from,
         }
 
     def save(self, path: Path):

{gptmed-0.4.0 → gptmed-0.4.1/gptmed.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gptmed
-Version: 0.4.0
+Version: 0.4.1
 Summary: A lightweight GPT-based language model framework for training custom question-answering models on any domain
 Author-email: Sanjog Sigdel <sigdelsanjog@gmail.com>
 Maintainer-email: Sanjog Sigdel <sigdelsanjog@gmail.com>

{gptmed-0.4.0 → gptmed-0.4.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "gptmed"
-version = "0.4.0"
+version = "0.4.1"
 description = "A lightweight GPT-based language model framework for training custom question-answering models on any domain"
 readme = "README.md"
 requires-python = ">=3.8"