Falcontune Docker images available

Applied Machine Learning Group, University of Waikato

2023-08-21 16:39

The falcontune library for fine-tuning and using Falcon 7B/40B models (which is based on the Hugging Face transformers library) is now available via the following docker images:

In-house registry:
- public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-huggingface-transformers:4.31.0_cuda11.7_falcontune_20230618
Docker hub:
- waikatodatamining/pytorch-huggingface-transformers:4.31.0_cuda11.7_falcontune_20230618