diff --git a/requirements/requirements-llm.txt b/requirements/requirements-llm.txt index 8935cb159..9bc21d251 100644 --- a/requirements/requirements-llm.txt +++ b/requirements/requirements-llm.txt @@ -1,3 +1,3 @@ # optimum-amd[brevitas] @ git+https://github.com/huggingface/optimum-amd.git@main tqdm -transformers +transformers[sentencepiece]==4.45.2 diff --git a/src/brevitas_examples/common/accelerate_utils/accelerate.py b/src/brevitas_examples/common/accelerate_utils/accelerate.py index 24c420f9f..ead616ed2 100644 --- a/src/brevitas_examples/common/accelerate_utils/accelerate.py +++ b/src/brevitas_examples/common/accelerate_utils/accelerate.py @@ -1,7 +1,6 @@ # Copyright (C) 2023, Advanced Micro Devices, Inc. All rights reserved. # SPDX-License-Identifier: BSD-3-Clause - import logging from typing import Dict, Mapping, Optional, Union