diff --git a/machine_learning_core/llm-awq/pyproject.toml b/machine_learning_core/llm-awq/pyproject.toml index d739ac22..085eaa5f 100644 --- a/machine_learning_core/llm-awq/pyproject.toml +++ b/machine_learning_core/llm-awq/pyproject.toml @@ -15,11 +15,11 @@ classifiers = [ dependencies = [ "accelerate", "sentencepiece", "tokenizers>=0.12.1", "torch>=2.0.0", "torchvision", - "transformers==4.36.2", + "transformers==4.38.0", "lm_eval==0.3.0", "texttable", "toml", "attributedict", "protobuf", - "gradio==3.35.2", "gradio_client==0.2.9", + "gradio==4.19.2", "gradio_client==0.2.9", "fastapi", "uvicorn", "pydantic==1.10.14" ]