We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ae1bf24 commit 82928bbCopy full SHA for 82928bb
tests/transformers/models/test_single_subfunction.py
@@ -11,6 +11,7 @@
11
from transformers import AutoConfig, AutoModelForCausalLM
12
13
from QEfficient.transformers.models.modeling_auto import QEFFAutoModelForCausalLM
14
+from QEfficient.utils.device_utils import get_available_device_id
15
16
torch.manual_seed(42)
17
0 commit comments