phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

RFC-0001-The-Llama-Stack (#8)

Browse source 
* RFC-0001-The-Llama-Stack

* Add OpenAPI generation utility, update SPEC to reflect latest types

* First cut at an observability API

* llama3_1 -> llama3

---------

Co-authored-by: Ashwin Bharambe <ashwin.bharambe@gmail.com>
This commit is contained in:
raghotham 2024-08-20 19:01:18 -07:00 • committed by GitHub
parent 57881c08c1
commit 2232bfa8b5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
19 changed files with 9177 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_toolchain/post_training/api/endpoints.py
View file

@ -95,13 +95,13 @@ class PostTrainingJobArtifactsResponse(BaseModel):
class PostTraining(Protocol):
@webmethod(route="/post_training/supervised_fine_tune")
def post_supervised_fine_tune(
def supervised_fine_tune(
self,
request: PostTrainingSFTRequest,
) -> PostTrainingJob: ...
@webmethod(route="/post_training/preference_optimize")
def post_preference_optimize(
def preference_optimize(
self,
request: PostTrainingRLHFRequest,
) -> PostTrainingJob: ...