Reduce a bunch of dependencies from toolchain

Some improvements to the distribution install script

llama_toolchain/inference/api/datatypes.py

@@ -7,9 +7,9 @@
 from enum import Enum
 from typing import List, Literal, Optional, Union
 
-from pydantic import BaseModel, Field
+from llama_models.schema_utils import json_schema_type
 
-from strong_typing.schema import json_schema_type
+from pydantic import BaseModel, Field
 from typing_extensions import Annotated
 
 from llama_models.llama3_1.api.datatypes import *  # noqa: F403

llama_toolchain/inference/api/endpoints.py

@@ -8,7 +8,7 @@ from .datatypes import *  # noqa: F403
 from typing import Optional, Protocol
 
 # this dependency is annoying and we need a forked up version anyway
-from pyopenapi import webmethod
+from llama_models.schema_utils import webmethod
 
 
 @json_schema_type

llama_toolchain/inference/meta_reference/config.py

@@ -6,8 +6,9 @@
 
 from typing import Optional
 
+from llama_models.schema_utils import json_schema_type
+
 from pydantic import BaseModel
-from strong_typing.schema import json_schema_type
 
 from llama_toolchain.inference.api import QuantizationConfig
 

llama_toolchain/inference/meta_reference/generation.py

@@ -25,8 +25,8 @@ from fairscale.nn.model_parallel.initialize import (
 from llama_models.llama3_1.api.args import ModelArgs
 from llama_models.llama3_1.api.chat_format import ChatFormat, ModelInput
 from llama_models.llama3_1.api.datatypes import Message
-from llama_models.llama3_1.api.model import Transformer
 from llama_models.llama3_1.api.tokenizer import Tokenizer
+from llama_models.llama3_1.reference_impl.model import Transformer
 from llama_models.sku_list import resolve_model
 from termcolor import cprint
 

llama_toolchain/inference/ollama/config.py

@@ -4,8 +4,8 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
+from llama_models.schema_utils import json_schema_type
 from pydantic import BaseModel, Field
-from strong_typing.schema import json_schema_type
 
 
 @json_schema_type