Browse Source

Move inference scripts into example folder

Matthias Reso 1 năm trước cách đây
mục cha
commit
ccda6fb8ca

src/llama_recipes/inference/README.md → examples/README.md


src/llama_recipes/inference/chat_completion.py → examples/chat_completion.py


src/llama_recipes/inference/chats.json → examples/chats.json


src/llama_recipes/inference/code_llama/__init__.py → examples/code_llama/__init__.py


src/llama_recipes/inference/code_llama/code_completion_example.py → examples/code_llama/code_completion_example.py


src/llama_recipes/inference/code_llama/code_completion_prompt.txt → examples/code_llama/code_completion_prompt.txt


src/llama_recipes/inference/code_llama/code_infilling_example.py → examples/code_llama/code_infilling_example.py


src/llama_recipes/inference/code_llama/code_infilling_prompt.txt → examples/code_llama/code_infilling_prompt.txt


src/llama_recipes/inference/hf_text_generation_inference/README.md → examples/hf_text_generation_inference/README.md


src/llama_recipes/inference/hf_text_generation_inference/__init__.py → examples/hf_text_generation_inference/__init__.py


src/llama_recipes/inference/hf_text_generation_inference/merge_lora_weights.py → examples/hf_text_generation_inference/merge_lora_weights.py


src/llama_recipes/inference/inference.py → examples/inference.py


src/llama_recipes/inference/samsum_prompt.txt → examples/samsum_prompt.txt


src/llama_recipes/inference/vLLM_inference.py → examples/vLLM_inference.py


+ 0 - 9
src/llama_recipes/inference/__main__.py

@@ -1,9 +0,0 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# This software may be used and distributed according to the terms of the Llama 2 Community License Agreement.
-
-import fire
-
-from llama_recipes.inference.inference import main
-
-if __name__ == "__main__":
-    fire.Fire(main)