phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 17:29:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

commit

Browse source
This commit is contained in:
Botao Chen 2025-01-02 12:16:49 -08:00
parent b438e616ff
commit 647c6312a8
2 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -15,6 +15,8 @@ from typing import Any, Callable, Dict, List
import torch import torch
from llama_models.datatypes import Model from llama_models.datatypes import Model
from llama_models.llama3.api.datatypes import BaseModel
from llama_models.sku_list import resolve_model from llama_models.sku_list import resolve_model
from llama_stack.apis.common.type_system import ParamType, StringType from llama_stack.apis.common.type_system import ParamType, StringType
from llama_stack.apis.datasets import Datasets from llama_stack.apis.datasets import Datasets

1
llama_stack/providers/inline/post_training/torchtune/recipes/lora_finetuning_single_device.py
View file

@ -7,6 +7,7 @@
import logging import logging
import os import os
import time import time
from datetime import datetime
from functools import partial from functools import partial
from pathlib import Path from pathlib import Path
from typing import Any, Dict, List, Optional, Tuple from typing import Any, Dict, List, Optional, Tuple