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 2828from QEfficient .peft import QEffAutoPeftModelForCausalLM
2929from QEfficient .transformers .transform import transform
3030from QEfficient .utils import custom_format_warning
31+ from QEfficient .utils .logging_utils import logger
3132
3233# custom warning for the better logging experience
3334warnings .formatwarning = custom_format_warning
5051 "QEFFAutoModelForSpeechSeq2Seq" ,
5152 "QEFFCommonLoader" ,
5253]
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