Skip to content

Commit

Permalink
remove mistral7b model config
Browse files Browse the repository at this point in the history
  • Loading branch information
Mithun Vanniasinghe committed Jan 20, 2025
1 parent c95b2d3 commit 06f73ef
Showing 1 changed file with 0 additions and 14 deletions.
14 changes: 0 additions & 14 deletions app/api/shared_config/model_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -198,20 +198,6 @@ def base_docker_config():
service_route="/v1/chat/completions",
env_file=os.environ.get("VLLM_LLAMA31_ENV_FILE"),
),
ModelImpl(
model_name="Mistral7B-instruct-v0.2",
model_id="id_tt-metal-mistral-7bv0.0.2",
image_name="ghcr.io/tenstorrent/tt-inference-server/tt-metal-mistral-7b-src-base",
image_tag="v0.0.3-tt-metal-v0.52.0-rc33",
hf_model_path="mistralai/Mistral-7B-Instruct-v0.2",
device_configurations={DeviceConfigurations.N300x4},
docker_config=base_docker_config(),
user_uid=1000,
user_gid=1000,
shm_size="32G",
service_port=7000,
service_route="/inference/mistral7b",
),
#! Add new model vLLM model implementations here
# ModelImpl(
# model_name="", #? Add the model name for the vLLM model based on persistent storage
Expand Down

0 comments on commit 06f73ef

Please sign in to comment.