Added logger warning

asmigosw · asmigosw · commit 95676581c351 · 2025-03-03T10:49:01.000Z
Signed-off-by: Asmita Goswami &lt;quic_asmigosw@quicinc.com&gt;
diff --git a/QEfficient/cloud/infer.py b/QEfficient/cloud/infer.py
@@ -168,6 +168,8 @@ def main(
         )
         print(output)
     else:
+        if image_path is not None or image_url is not None:
+            logger.warning(f"Skipping Image as it is not acceptable argument for {architecture}")
         tokenizer = load_hf_tokenizer(
             pretrained_model_name_or_path=(local_model_dir if local_model_dir else model_name),
             cache_dir=cache_dir,

Original file line number	Diff line number	Diff line change
`@@ -168,6 +168,8 @@ def main(`
`168`	`168`	`)`
`169`	`169`	`print(output)`
`170`	`170`	`else:`
	`171`	`+ if image_path is not None or image_url is not None:`
	`172`	`+ logger.warning(f"Skipping Image as it is not acceptable argument for {architecture}")`
`171`	`173`	`tokenizer = load_hf_tokenizer(`
`172`	`174`	`pretrained_model_name_or_path=(local_model_dir if local_model_dir else model_name),`
`173`	`175`	`cache_dir=cache_dir,`