Merge remote-tracking branch 'origin/main' into support_3pt1_8b

This commit is contained in:
Botao Chen 2025-01-02 15:22:25 -08:00
commit 8cdc91e619
52 changed files with 673 additions and 226 deletions

View file

@ -18,6 +18,7 @@ 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
from torchtune.models.llama3 import llama3_tokenizer

View file

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