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

Add init files to post training folders (#711)

Browse source 
add init files to post training folders to make pkg build pick up those
files

## Test
WIP colab notebook
https://colab.research.google.com/drive/1K4Q2wZq232_Bpy2ud4zL9aRxvCWAwyQs?usp=sharing
to sharecase the post training APIs
This commit is contained in:
Botao Chen 2025-01-13 20:19:18 -08:00 • committed by GitHub
parent f320eede2b
commit 747683a8a2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 41 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/providers/inline/post_training/torchtune/common/__init__.py Normal file
View file

@ -0,0 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.

5
llama_stack/providers/inline/post_training/torchtune/common/utils.py
View file

@ -16,8 +16,6 @@ from typing import Any, Callable, Dict, List
import torch
from llama_models.datatypes import Model
from llama_models.sku_list import resolve_model
from llama_stack.apis.common.type_system import ParamType, StringType
from llama_stack.apis.datasets import Datasets
from pydantic import BaseModel
@ -26,6 +24,9 @@ from torchtune.models.llama3._tokenizer import Llama3Tokenizer
from torchtune.models.llama3_1 import lora_llama3_1_8b
from torchtune.models.llama3_2 import lora_llama3_2_3b
from llama_stack.apis.common.type_system import ParamType, StringType
from llama_stack.apis.datasets import Datasets
class ColumnName(Enum):
instruction = "instruction"