update

intel-analytics · Jun 20, 2024 · 669dd69 · 669dd69
1 parent 699d97a
commit 669dd69
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/llm/example/CPU/PyTorch-Models/Model/llava/generate.py b/python/llm/example/CPU/PyTorch-Models/Model/llava/generate.py
@@ -272,6 +272,9 @@ def get_stopping_criteria(conv, tokenizer, input_ids):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(
+
+
+if __name__ == '__main__':
         description='Predict Tokens using `generate()` API for LLaVA model')
     parser.add_argument('--repo-id-or-model-path', type=str, default="liuhaotian/llava-v1.5-13b",
                         help='The huggingface repo id for the LLaVA model to be downloaded'