huggingface / huggingface/evaluate

Evaluate a quantize model

Open
#487 2 comments 7 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
2.5k
Forks
341
PR merge metrics
No merged PRs in 30d

Description

I am trying to run an evaluation in a model that is quantized. I have to instantiate it using 'accelerate' to use the GPU, because otherwise it cannot fit in memory. The problem is that when using compute() in the metric I want, I get this error:

`ValueError: The model has been loaded with `accelerate` and therefore cannot be moved to a specific device. Please discard the `device` argument when creating your pipeline object.`

Is there any way to allow using the compute() method in a model that is in GPU?

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.