From b6b7d7c3c415d67976e054ab5ad5be6b5d5b460d Mon Sep 17 00:00:00 2001 From: Kaihui-intel Date: Thu, 26 Sep 2024 17:21:54 +0800 Subject: [PATCH] Update auto_round requirements for transformers example (#2013) Signed-off-by: Kaihui-intel --- .../weight_only/text-generation/requirements_GPU.txt | 2 +- .../weight_only/text-generation/requirements_cpu_woq.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_GPU.txt b/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_GPU.txt index 3d5e737d101..fe091ca0ef0 100644 --- a/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_GPU.txt +++ b/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_GPU.txt @@ -11,5 +11,5 @@ bitsandbytes #baichuan transformers_stream_generator tiktoken #qwen einops #qwen -autoround +auto_round lm-eval==0.4.3 diff --git a/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt b/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt index d260c5e3b8b..63f7c30d695 100644 --- a/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt +++ b/examples/3.x_api/pytorch/nlp/huggingface_models/language-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt @@ -10,6 +10,6 @@ bitsandbytes #baichuan transformers_stream_generator tiktoken #qwen einops #qwen -autoround +auto_round lm-eval==0.4.3 huggingface_hub