diff --git a/easybuild/easyconfigs/a/accelerate/accelerate-1.10.0-foss-2024a.eb b/easybuild/easyconfigs/a/accelerate/accelerate-1.10.0-foss-2024a.eb new file mode 100644 index 00000000000..104cf85722b --- /dev/null +++ b/easybuild/easyconfigs/a/accelerate/accelerate-1.10.0-foss-2024a.eb @@ -0,0 +1,30 @@ +easyblock = 'PythonBundle' + +name = 'accelerate' +version = '1.10.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': '2024a'} + +dependencies = [ + ('Python', '3.12.3'), + ('SciPy-bundle', '2024.05'), + ('PyTorch', '2.6.0'), + ('PyYAML', '6.0.2'), + ('Safetensors', '0.6.2'), + ('huggingface_hub', '0.34.4'), +] + +exts_list = [ + (name, version, { + 'checksums': ['8270568fda9036b5cccdc09703fef47872abccd56eb5f6d53b54ea5fb7581496'], + }), +] + +sanity_check_commands = ['accelerate test'] + +moduleclass = 'ai'