Update infer and execute API

Signed-off-by: mamtsing <[email protected]>
quic · May 17, 2024 · a9a9ed0 · a9a9ed0
1 parent 861e60c
commit a9a9ed0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/QEfficient/cloud/execute.py b/QEfficient/cloud/execute.py
@@ -78,6 +78,7 @@ def main(
     )
     parser.add_argument(
         "--inputs_file_path",
+        "--inputs-file-path",
         type=str,
         help="for batch size>1, pass input prompts in txt file, sample prompts.txt file present in examples folder",
     )

diff --git a/QEfficient/cloud/infer.py b/QEfficient/cloud/infer.py
@@ -189,7 +189,7 @@ def main(
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
-        description="Inference command, the model will be downloaded from HF, optmized, compiled, executed on AIC"
+        description="Inference command, the model will be downloaded from HF, optmized, compiled, executed on Cloud AI 100"
     )
     parser.add_argument("--model-name", "--model_name", required=True, help="HF Model card name/id")
     parser.add_argument(