arbor-ai 0.2.2__tar.gz → 0.2.3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (57) hide show
  1. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/PKG-INFO +10 -2
  2. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/README.md +8 -0
  3. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor_ai.egg-info/PKG-INFO +10 -2
  4. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/pyproject.toml +2 -2
  5. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/LICENSE +0 -0
  6. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/__init__.py +0 -0
  7. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/cli.py +0 -0
  8. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/client/__init__.py +0 -0
  9. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/client/api.py +0 -0
  10. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/client/arbor_client.py +0 -0
  11. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/__init__.py +0 -0
  12. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/api/__init__.py +0 -0
  13. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/api/models/schemas.py +0 -0
  14. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/api/routes/__init__.py +0 -0
  15. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/api/routes/files.py +0 -0
  16. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/api/routes/grpo.py +0 -0
  17. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/api/routes/inference.py +0 -0
  18. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/api/routes/jobs.py +0 -0
  19. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/core/__init__.py +0 -0
  20. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/core/config.py +0 -0
  21. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/core/config_manager.py +0 -0
  22. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/core/logging.py +0 -0
  23. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/main.py +0 -0
  24. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/__init__.py +0 -0
  25. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/comms/__init__.py +0 -0
  26. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/comms/comms.py +0 -0
  27. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/dependencies.py +0 -0
  28. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/file_manager.py +0 -0
  29. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/grpo_manager.py +0 -0
  30. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/health_manager.py +0 -0
  31. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/inference/__init__.py +0 -0
  32. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/inference/vllm_client.py +0 -0
  33. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/inference/vllm_serve.py +0 -0
  34. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/inference_manager.py +0 -0
  35. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/job_manager.py +0 -0
  36. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/dpo_training.py +0 -0
  37. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/grpo_training.py +0 -0
  38. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/mmgrpo_training.py +0 -0
  39. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/sft_training.py +0 -0
  40. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/utils/__init__.py +0 -0
  41. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/utils/arg_parser.py +0 -0
  42. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/utils/callbacks.py +0 -0
  43. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/utils/comms_monitors.py +0 -0
  44. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/utils/dataset.py +0 -0
  45. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/utils/ingestion_monitor.py +0 -0
  46. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/scripts/utils/mock_server.py +0 -0
  47. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/services/training_manager.py +0 -0
  48. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/utils/__init__.py +0 -0
  49. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/utils/helpers.py +0 -0
  50. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor/server/utils/logging.py +0 -0
  51. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor_ai.egg-info/SOURCES.txt +0 -0
  52. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor_ai.egg-info/dependency_links.txt +0 -0
  53. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor_ai.egg-info/entry_points.txt +0 -0
  54. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor_ai.egg-info/requires.txt +0 -0
  55. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/arbor_ai.egg-info/top_level.txt +0 -0
  56. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/setup.cfg +0 -0
  57. {arbor_ai-0.2.2 → arbor_ai-0.2.3}/tests/test_cli.py +0 -0
@@ -1,11 +1,11 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: arbor-ai
3
- Version: 0.2.2
3
+ Version: 0.2.3
4
4
  Summary: A framework for fine-tuning and managing language models
5
5
  Author-email: Noah Ziems <nziems2@nd.edu>
6
6
  Project-URL: Homepage, https://github.com/Ziems/arbor
7
7
  Project-URL: Issues, https://github.com/Ziems/arbor/issues
8
- Requires-Python: >=3.10
8
+ Requires-Python: >=3.11
9
9
  Description-Content-Type: text/markdown
10
10
  License-File: LICENSE
11
11
  Requires-Dist: torch>=2.6.0
@@ -79,6 +79,14 @@ export NCCL_P2P_DISABLE=1
79
79
  export NCCL_IB_DISABLE=1
80
80
  ```
81
81
 
82
+ **NVCC**
83
+ If you run into issues, double check that you have [nvcc](https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/) installed:
84
+ ```bash
85
+ nvcc --version
86
+ ```
87
+ If you don't have admin permissions, you can often install nvcc using conda.
88
+
89
+
82
90
  ## 🙏 Acknowledgements
83
91
 
84
92
  Arbor builds on the shoulders of great work. We extend our thanks to:
@@ -52,6 +52,14 @@ export NCCL_P2P_DISABLE=1
52
52
  export NCCL_IB_DISABLE=1
53
53
  ```
54
54
 
55
+ **NVCC**
56
+ If you run into issues, double check that you have [nvcc](https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/) installed:
57
+ ```bash
58
+ nvcc --version
59
+ ```
60
+ If you don't have admin permissions, you can often install nvcc using conda.
61
+
62
+
55
63
  ## 🙏 Acknowledgements
56
64
 
57
65
  Arbor builds on the shoulders of great work. We extend our thanks to:
@@ -1,11 +1,11 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: arbor-ai
3
- Version: 0.2.2
3
+ Version: 0.2.3
4
4
  Summary: A framework for fine-tuning and managing language models
5
5
  Author-email: Noah Ziems <nziems2@nd.edu>
6
6
  Project-URL: Homepage, https://github.com/Ziems/arbor
7
7
  Project-URL: Issues, https://github.com/Ziems/arbor/issues
8
- Requires-Python: >=3.10
8
+ Requires-Python: >=3.11
9
9
  Description-Content-Type: text/markdown
10
10
  License-File: LICENSE
11
11
  Requires-Dist: torch>=2.6.0
@@ -79,6 +79,14 @@ export NCCL_P2P_DISABLE=1
79
79
  export NCCL_IB_DISABLE=1
80
80
  ```
81
81
 
82
+ **NVCC**
83
+ If you run into issues, double check that you have [nvcc](https://docs.nvidia.com/cuda/cuda-compiler-driver-nvcc/) installed:
84
+ ```bash
85
+ nvcc --version
86
+ ```
87
+ If you don't have admin permissions, you can often install nvcc using conda.
88
+
89
+
82
90
  ## 🙏 Acknowledgements
83
91
 
84
92
  Arbor builds on the shoulders of great work. We extend our thanks to:
@@ -1,12 +1,12 @@
1
1
  [project]
2
2
  name = "arbor-ai"
3
- version = "0.2.2"
3
+ version = "0.2.3"
4
4
  description = "A framework for fine-tuning and managing language models"
5
5
  authors = [
6
6
  {name = "Noah Ziems", email = "nziems2@nd.edu"}
7
7
  ]
8
8
  readme = "README.md"
9
- requires-python = ">=3.10"
9
+ requires-python = ">=3.11"
10
10
  dependencies = [
11
11
  "torch>=2.6.0",
12
12
  "fastapi",
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes