diff --git a/easybuild/easyconfigs/a/accelerate/accelerate-1.12.0-foss-2025a-CUDA-12.8.0.eb b/easybuild/easyconfigs/a/accelerate/accelerate-1.12.0-foss-2025a-CUDA-12.8.0.eb new file mode 100644 index 000000000000..e2393e55f772 --- /dev/null +++ b/easybuild/easyconfigs/a/accelerate/accelerate-1.12.0-foss-2025a-CUDA-12.8.0.eb @@ -0,0 +1,31 @@ +easyblock = 'PythonBundle' + +name = 'accelerate' +version = '1.12.0' +versionsuffix = '-CUDA-%(cudaver)s' + +homepage = 'https://github.com/huggingface/accelerate' +description = """A simple way to launch, train, and use PyTorch models on almost any device and +distributed configuration, automatic mixed precision (including fp8), +and easy-to-configure FSDP and DeepSpeed support.""" + +toolchain = {'name': 'foss', 'version': '2025a'} + +dependencies = [ + ('CUDA', '12.8.0', '', SYSTEM), + ('Python', '3.13.1'), + ('PyTorch', '2.9.1', f"{versionsuffix}-whl"), + ('PyYAML', '6.0.2'), + ('Safetensors', '0.7.0'), + ('huggingface-hub', '1.2.4'), +] + +exts_list = [ + (name, version, { + 'checksums': ['70988c352feb481887077d2ab845125024b2a137a5090d6d7a32b57d03a45df6'], + }), +] + +sanity_check_commands = ['accelerate test'] + +moduleclass = 'ai' diff --git a/easybuild/easyconfigs/a/accelerate/accelerate-1.12.0-foss-2025a.eb b/easybuild/easyconfigs/a/accelerate/accelerate-1.12.0-foss-2025a.eb new file mode 100644 index 000000000000..1c402a30b289 --- /dev/null +++ b/easybuild/easyconfigs/a/accelerate/accelerate-1.12.0-foss-2025a.eb @@ -0,0 +1,29 @@ +easyblock = 'PythonBundle' + +name = 'accelerate' +version = '1.12.0' + +homepage = 'https://github.com/huggingface/accelerate' +description = """A simple way to launch, train, and use PyTorch models on almost any device and +distributed configuration, automatic mixed precision (including fp8), +and easy-to-configure FSDP and DeepSpeed support.""" + +toolchain = {'name': 'foss', 'version': '2025a'} + +dependencies = [ + ('Python', '3.13.1'), + ('PyTorch', '2.9.1', '-whl'), + ('PyYAML', '6.0.2'), + ('Safetensors', '0.7.0'), + ('huggingface-hub', '1.2.4'), +] + +exts_list = [ + (name, version, { + 'checksums': ['70988c352feb481887077d2ab845125024b2a137a5090d6d7a32b57d03a45df6'], + }), +] + +sanity_check_commands = ['accelerate test'] + +moduleclass = 'ai'