Skip to content

Commit

Permalink
drop to gfbf toolchain
Browse files Browse the repository at this point in the history
  • Loading branch information
Samuel Moors committed Jun 17, 2024
1 parent f71916b commit b6d318a
Showing 1 changed file with 1 addition and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@ easyblock = "CargoPythonBundle"

name = 'Safetensors'
version = '0.4.3'
versionsuffix = '-CUDA-%(cudaver)s'
_rustver = '1.75.0'

homepage = 'https://huggingface.co/docs/safetensors'
Expand All @@ -12,20 +11,16 @@ pickle) and that is still fast (zero-copy). Safetensors is really fast.
This variant of Safetensors is installed with support for numpy and PyTorch
"""

toolchain = {'name': 'foss', 'version': '2023a'}
toolchain = {'name': 'gfbf', 'version': '2023a'}

builddependencies = [
('Rust', _rustver),
('maturin', '1.4.0', '-Rust-' + _rustver),
]

dependencies = [
('CUDA', '12.1.1', '', SYSTEM),
('Python', '3.11.3'),
('SciPy-bundle', '2023.07'),
('PyTorch', '2.1.2', versionsuffix), # optional
# ('jax', 'xxx', versionsuffix), # optional, add 'jax' to 'use_pip_extras'
# ('TensorFlow', 'xxx', versionsuffix), # optional, add 'tensorflow' to 'use_pip_extras'
]

# crates generated on 2024-05-24 from directories savetensors/ and bindings/python/
Expand Down Expand Up @@ -291,7 +286,6 @@ use_pip = True

exts_list = [
('safetensors', version, {
'use_pip_extras': 'numpy,torch',
'checksums': ['2f85fc50c4e07a21e95c24e07460fe6f7e2859d0ce88092838352b798ce711c2'],
}),
]
Expand Down

0 comments on commit b6d318a

Please sign in to comment.