Reduce a bunch of dependencies from toolchain

Some improvements to the distribution install script
This commit is contained in:
Ashwin Bharambe 2024-08-07 18:02:35 -07:00
parent 171a178783
commit f27d629fe8
27 changed files with 82 additions and 103 deletions

View file

@ -9,9 +9,9 @@ from typing import Dict, Optional
from llama_models.llama3_1.api.datatypes import URL
from pydantic import BaseModel
from llama_models.schema_utils import json_schema_type
from strong_typing.schema import json_schema_type
from pydantic import BaseModel
@json_schema_type

View file

@ -5,8 +5,8 @@
# the root directory of this source tree.
from llama_models.llama3_1.api.datatypes import URL
from llama_models.schema_utils import json_schema_type
from pydantic import BaseModel
from strong_typing.schema import json_schema_type
@json_schema_type(schema={"description": "Checkpoint created during training runs"})