-
|
Can this framework used for mutil-gpus and work in tensor parallel |
Beta Was this translation helpful? Give feedback.
Answered by
SimJeg
Mar 3, 2025
Replies: 1 comment
-
|
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 |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
SimJeg
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi,
kvpress is compatible with huggingface accelerate for multi-GPU settings:
I will add it to the README FAQ section