huggingface / huggingface/accelerate

[feature request] INT4 inference support and flexgen like offloading.

Open
#1,157 0 comments 10 reactions 0 assignees View on GitHub
feature request
Dominant language
Python
Stars
9.9k
Forks
1.5k
Avg merge
5d 2h
Merged PRs (30d)
27

Description

Hello.

INT4 is accelerated on Ada Lovelace, Ampere and Turing GPU architectures tensor cores and can effectively halve VRAM requirements compared to INT8 (and that halves memory consumption in comparison to FP16), allowing for larger language models to easily run on consumer GPU hardware. According to the paper https://arxiv.org/abs/2212.09720 The accuracy loss is minimal.

There's also a new off-loading system called FlexGen which drastically improves speed compared to HF accelerate when offloading to the CPU. https://github.com/FMInference/FlexGen

I wish for Huggingface to implement these innovations into HF accelerate natively, so that LLMs can be run on consumer GPUs in an user-friendly, performant and intuitive way. Thank you.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.