llmflowstack 1.3.0__tar.gz → 1.3.1__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 (28) hide show
  1. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/PKG-INFO +3 -2
  2. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/pyproject.toml +3 -2
  3. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/.github/workflows/python-publish.yml +0 -0
  4. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/.gitignore +0 -0
  5. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/LICENSE +0 -0
  6. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/README.md +0 -0
  7. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/__init__.py +0 -0
  8. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/callbacks/__init__.py +0 -0
  9. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/callbacks/force_json.py +0 -0
  10. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/callbacks/log_collector.py +0 -0
  11. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/collators/multimodal_causal.py +0 -0
  12. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/decoders/__init__.py +0 -0
  13. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/decoders/base_decoder.py +0 -0
  14. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/decoders/gemma_3.py +0 -0
  15. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/decoders/gpt_2.py +0 -0
  16. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/decoders/gpt_oss.py +0 -0
  17. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/decoders/llama_3.py +0 -0
  18. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/decoders/llama_4.py +0 -0
  19. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/decoders/medgemma.py +0 -0
  20. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/decoders/qwen_3.py +0 -0
  21. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/rag/VectorDatabase.py +0 -0
  22. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/rag/__init__.py +0 -0
  23. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/schemas/__init__.py +0 -0
  24. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/schemas/params.py +0 -0
  25. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/utils/__init__.py +0 -0
  26. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/utils/evaluation_methods.py +0 -0
  27. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/utils/exceptions.py +0 -0
  28. {llmflowstack-1.3.0 → llmflowstack-1.3.1}/llmflowstack/utils/logging.py +0 -0
{llmflowstack-1.3.0 → llmflowstack-1.3.1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: llmflowstack
3
- Version: 1.3.0
3
+ Version: 1.3.1
4
4
  Summary: LLMFlowStack is a framework for training and using LLMs (LLaMA, GPT-OSS, Gemma, ...). Supports DAPT, fine-tuning, and distributed inference.
5
5
  Author-email: Gustavo Henrique Ferreira Cruz <gustavohferreiracruz@gmail.com>
6
6
  License: MIT
@@ -15,6 +15,7 @@ Requires-Dist: fbgemm-gpu-genai
15
15
  Requires-Dist: kernels
16
16
  Requires-Dist: langchain-chroma
17
17
  Requires-Dist: langchain-community
18
+ Requires-Dist: mslk-cuda==1.0.0
18
19
  Requires-Dist: nltk
19
20
  Requires-Dist: peft
20
21
  Requires-Dist: pillow
@@ -22,7 +23,7 @@ Requires-Dist: rouge-score
22
23
  Requires-Dist: safetensors
23
24
  Requires-Dist: sentence-transformers
24
25
  Requires-Dist: torch
25
- Requires-Dist: torchao
26
+ Requires-Dist: torchao==0.16.0
26
27
  Requires-Dist: transformers==4.57.6
27
28
  Requires-Dist: triton
28
29
  Requires-Dist: trl
{llmflowstack-1.3.0 → llmflowstack-1.3.1}/pyproject.toml RENAMED
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
4
4
 
5
5
  [project]
6
6
  name = "llmflowstack"
7
- version = "1.3.0"
7
+ version = "1.3.1"
8
8
  authors = [
9
9
  { name = "Gustavo Henrique Ferreira Cruz", email = "gustavohferreiracruz@gmail.com" }
10
10
  ]
@@ -23,6 +23,7 @@ dependencies = [
23
23
  "kernels",
24
24
  "langchain-chroma",
25
25
  "langchain_community",
26
+ "mslk-cuda==1.0.0",
26
27
  "nltk",
27
28
  "peft",
28
29
  "pillow",
@@ -30,7 +31,7 @@ dependencies = [
30
31
  "safetensors",
31
32
  "sentence-transformers",
32
33
  "torch",
33
- "torchao",
34
+ "torchao==0.16.0",
34
35
  "transformers==4.57.6",
35
36
  "triton",
36
37
  "trl",
{llmflowstack-1.3.0 → llmflowstack-1.3.1}/.gitignore RENAMED
File without changes
{llmflowstack-1.3.0 → llmflowstack-1.3.1}/LICENSE RENAMED
File without changes
{llmflowstack-1.3.0 → llmflowstack-1.3.1}/README.md RENAMED
File without changes