broccoli-ml 6.0.0__tar.gz → 6.0.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 (13) hide show
  1. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/PKG-INFO +1 -1
  2. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/transformer.py +1 -0
  3. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/pyproject.toml +1 -1
  4. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/LICENSE +0 -0
  5. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/README.md +0 -0
  6. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/__init__.py +0 -0
  7. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/activation.py +0 -0
  8. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/cnn.py +0 -0
  9. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/linear.py +0 -0
  10. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/rope.py +0 -0
  11. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/tensor.py +0 -0
  12. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/utils.py +0 -0
  13. {broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/vit.py +0 -0
{broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: broccoli-ml
3
- Version: 6.0.0
3
+ Version: 6.0.1
4
4
  Summary: Some useful Pytorch models, circa 2025
5
5
  License: MIT
6
6
  Author: Nicholas Bailey
{broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/transformer.py RENAMED
@@ -13,6 +13,7 @@ from .rope import RotaryEmbedding, apply_rotary_emb
13
13
  try:
14
14
  from flash_attn import flash_attn_func
15
15
 
16
+ print("Using flash-attn.")
16
17
  FLASH_ATTN = True
17
18
  except ImportError:
18
19
  pass
{broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "broccoli-ml"
3
- version = "6.0.0"
3
+ version = "6.0.1"
4
4
  description = "Some useful Pytorch models, circa 2025"
5
5
  authors = [
6
6
  {name = "Nicholas Bailey"}
{broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/LICENSE RENAMED
File without changes
{broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/README.md RENAMED
File without changes
{broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/cnn.py RENAMED
File without changes
{broccoli_ml-6.0.0 → broccoli_ml-6.0.1}/broccoli/vit.py RENAMED
File without changes