phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-18 12:59:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix pre-commit

Browse source
This commit is contained in:
Botao Chen 2024-12-02 17:59:53 -08:00
parent 745163b6b0
commit 5838b7211d
3 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/inline/post_training/meta_reference/datasets/sft.py
View file

@ -1,3 +1,9 @@
# 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.
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#

5
llama_stack/providers/inline/post_training/meta_reference/recipes/lora_finetuning_single_device.py
View file

@ -4,20 +4,17 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
import asyncio
import logging
import os
import re
from functools import partial
from pathlib import Path
from typing import Any, Dict, List, Optional, Tuple, Union
from typing import Any, Dict, List, Optional, Tuple
import torch
from llama_models.sku_list import resolve_model
from llama_stack.apis.datasetio import DatasetIO
from torch import nn
from torchtune import utils as torchtune_utils
from torchtune.training.checkpointing._utils import ModelType
from llama_stack.apis.post_training import * # noqa
from llama_stack.apis.post_training import PostTrainingSFTRequest
from llama_stack.distribution.utils.model_utils import model_local_dir