mostlyai-engine 1.0.2__tar.gz → 1.0.3__tar.gz

{mostlyai_engine-1.0.2 → mostlyai_engine-1.0.3}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: mostlyai-engine
-Version: 1.0.2
-Summary: Synthetic Data - Engine
+Version: 1.0.3
+Summary: Synthetic Data Engine
 Project-URL: homepage, https://github.com/mostly-ai/mostlyai-engine
 Project-URL: repository, https://github.com/mostly-ai/mostlyai-engine
 Project-URL: documentation, https://mostly-ai.github.io/mostlyai-engine/
@@ -38,14 +38,14 @@ Requires-Dist: pyarrow>=16.0.0
 Requires-Dist: rich<14,>=13.9.4
 Requires-Dist: setuptools<76,>=75.0.0
 Requires-Dist: tokenizers<0.21,>=0.20.1
-Requires-Dist: torch==2.5.1; sys_platform == 'darwin'
+Requires-Dist: torch>=2.5.1; sys_platform == 'darwin'
 Requires-Dist: transformers<5,>=4.45.2
 Provides-Extra: cpu
-Requires-Dist: torch==2.5.1+cpu; (sys_platform != 'darwin') and extra == 'cpu'
-Requires-Dist: torch==2.5.1; (sys_platform == 'darwin') and extra == 'cpu'
+Requires-Dist: torch==2.5.1+cpu; (sys_platform == 'linux') and extra == 'cpu'
+Requires-Dist: torch>=2.5.1; (sys_platform != 'linux') and extra == 'cpu'
 Provides-Extra: gpu
 Requires-Dist: bitsandbytes<0.44,>=0.43.3; extra == 'gpu'
-Requires-Dist: torch==2.5.1; extra == 'gpu'
+Requires-Dist: torch>=2.5.1; extra == 'gpu'
 Requires-Dist: vllm<0.7,>=0.6.4; extra == 'gpu'
 Description-Content-Type: text/markdown
 

{mostlyai_engine-1.0.2 → mostlyai_engine-1.0.3}/mostlyai/engine/__init__.py

@@ -22,7 +22,7 @@ from mostlyai.engine.generation import generate
 
 
 __all__ = ["split", "analyze", "encode", "train", "generate", "init_logging"]
-__version__ = "1.0.2"
+__version__ = "1.0.3"
 
 # suppress specific warning related to os.fork() in multi-threaded processes
 warnings.filterwarnings("ignore", category=DeprecationWarning, message=".*multi-threaded.*fork.*")

{mostlyai_engine-1.0.2 → mostlyai_engine-1.0.3}/mostlyai/engine/_language/generation.py

@@ -210,7 +210,7 @@ def generate(
             False  # needed for gated hf models that are not sharded, otherwise GatedRepoError in vLLM
         )
         # set the default env var so that we don't pass it explicitly to vLLM
-        os.environ["HF_TOKEN"] = os.getenv("MOSTLY_HUGGING_FACE_TOKEN", "")
+        os.environ["HF_TOKEN"] = os.getenv("MOSTLY_HUGGING_FACE_TOKEN") or os.getenv("HF_TOKEN", "")
 
         is_peft_adapter = (workspace.model_path / "adapter_config.json").exists()
         if is_peft_adapter and device.type == "cuda":

{mostlyai_engine-1.0.2 → mostlyai_engine-1.0.3}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "mostlyai-engine"
-version = "1.0.2"
-description = "Synthetic Data - Engine"
+version = "1.0.3"
+description = "Synthetic Data Engine"
 authors = [{ name = "MOSTLY AI", email = "dev@mostly.ai" }]
 requires-python = ">=3.10"
 readme = "README.md"
@@ -28,7 +28,6 @@ dependencies = [
     "setuptools>=75.0.0,<76",
     "numpy>=1.26.3,<2",
     "pandas~=2.2.0",
-    "torch==2.5.1; sys_platform == 'darwin'",
     "pyarrow>=16.0.0",
     "joblib>=1.3.0,<2",
     "psutil>=5.9.5,<6",  # upgrade when colab psutil is updated
@@ -41,17 +40,18 @@ dependencies = [
     "formatron>=0.4.11,<0.5",
     "json-repair>=0.30.0,<0.31",
     "rich>=13.9.4,<14",
+    "torch>=2.5.1; sys_platform == 'darwin'",
 ]
 
 [project.optional-dependencies]
 gpu = [
-    "torch==2.5.1",
+    "torch>=2.5.1",
     "bitsandbytes>=0.43.3,<0.44",
     "vllm>=0.6.4,<0.7",
 ]
 cpu = [
-    "torch==2.5.1+cpu; sys_platform != 'darwin'",
-    "torch==2.5.1; sys_platform == 'darwin'"
+    "torch==2.5.1+cpu; sys_platform == 'linux'",
+    "torch>=2.5.1; sys_platform != 'linux'",
 ]
 
 [dependency-groups]
@@ -93,7 +93,7 @@ explicit = true
 
 [tool.uv.sources]
 torch = [
-  { index = "pytorch-cpu", extra = "cpu", marker = "sys_platform != 'darwin'" },
+  { index = "pytorch-cpu", extra = "cpu", marker = "sys_platform == 'linux'"},
 ]
 
 [tool.hatch.build.targets.sdist]