Skip to content
Discussion options

You must be logged in to vote

Hi,

kvpress is compatible with huggingface accelerate for multi-GPU settings:

pipe = pipeline("kv-press-text-generation", model=model, device_map="auto")

I will add it to the README FAQ section

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by SimJeg
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants