stanford-crfm / helm

Holistic Evaluation of Language Models (HELM), a framework to increase the transparency of language models (https://arxiv.org/abs/2211.09110). This framework is also used to evaluate text-to-image models in Holistic Evaluation of Text-to-Image Models (HEIM) (https://arxiv.org/abs/2311.04287).
https://crfm.stanford.edu/helm
Apache License 2.0
1.89k stars 243 forks source link

Add multi-GPU support to HuggingFaceClient #2762

Closed yifanmai closed 3 months ago

yifanmai commented 3 months ago

Add multi-GPU support to HuggingFaceClient via Hugging Face Accelerate. To use multi-GPU support, set device_map to auto in the model deployment.

Also enable multi-GPU support for some large Hugging Face models.