Updates to setup and requirements for PyPI

README.md

@@ -3,7 +3,12 @@
 This repo contains the API specifications for various components of the Llama Stack as well implementations for some of those APIs like model inference.
 The Stack consists of toolchain-apis and agentic-apis. This repo contains the toolchain-apis
 
-## Installation and Setup ## 
+## Installation
+
+You can install this repository as a [package](https://pypi.org/project/llama-toolchain/) by just doing `pip install llama-toolchain`
+
+If you want to install from source:
+
 ```bash
 mkdir -p ~/local
 cd ~/local
@@ -17,6 +22,7 @@ pip install -e .
 ```
 
 ## Test with cli
+
 We have built a llama cli to make it easy to configure / run parts of the toolchain
 ```
 llama --help

requirements.txt

@@ -6,11 +6,12 @@ fairscale
 fastapi
 fire
 flake8
-huggingface-hub
 httpx
+huggingface-hub
 hydra-core
 hydra-zen
 json-strong-typing
+llama_models
 matplotlib
 omegaconf
 pandas
@@ -28,5 +29,3 @@ ufmt==2.7.0
 usort==1.0.8
 uvicorn
 zmq
-
-llama_models[llama3_1] @ git+ssh://git@github.com/meta-llama/llama-models.git

setup.py

@@ -16,7 +16,7 @@ def read_requirements():
 
 setup(
     name="llama_toolchain",
-    version="0.0.0.1",
+    version="0.0.1",
     author="Meta Llama",
     author_email="llama-oss@meta.com",
     description="Llama toolchain",