File tree Expand file tree Collapse file tree 1 file changed +19
-0
lines changed Expand file tree Collapse file tree 1 file changed +19
-0
lines changed Original file line number Diff line number Diff line change 28
28
from QEfficient .peft import QEffAutoPeftModelForCausalLM
29
29
from QEfficient .transformers .transform import transform
30
30
from QEfficient .utils import custom_format_warning
31
+ from QEfficient .utils .logging_utils import logger
31
32
32
33
# custom warning for the better logging experience
33
34
warnings .formatwarning = custom_format_warning
50
51
"QEFFAutoModelForSpeechSeq2Seq" ,
51
52
"QEFFCommonLoader" ,
52
53
]
54
+
55
+
56
+ def check_qaic_sdk ():
57
+ """Check if QAIC SDK is installed"""
58
+ try :
59
+ import platform
60
+ import sys
61
+
62
+ sys .path .append (f"/opt/qti-aic/dev/lib/{ platform .machine ()} " )
63
+ import qaicrt # noqa: F401
64
+
65
+ return True
66
+ except ImportError :
67
+ return False
68
+
69
+
70
+ if not check_qaic_sdk ():
71
+ logger .warning ("QAIC SDK is not installed, eager mode features won't be available!" )
You can’t perform that action at this time.
0 commit comments