Skip to content

Commit 82928bb

Browse files
lint
Signed-off-by: abhishek-singh591 <sabhis@qti.qualcomm.com>
1 parent ae1bf24 commit 82928bb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/transformers/models/test_single_subfunction.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from transformers import AutoConfig, AutoModelForCausalLM
1212

1313
from QEfficient.transformers.models.modeling_auto import QEFFAutoModelForCausalLM
14+
from QEfficient.utils.device_utils import get_available_device_id
1415

1516
torch.manual_seed(42)
1617

0 commit comments

Comments
 (0)