Update auto_round requirements for transformers example (#2013)

Signed-off-by: Kaihui-intel <[email protected]>
intel · Sep 26, 2024 · b6b7d7c · b6b7d7c
1 parent ee600ba
commit b6b7d7c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...guage-modeling/quantization/transformers/weight_only/text-generation/requirements_GPU.txt b/...guage-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/...e-modeling/quantization/transformers/weight_only/text-generation/requirements_cpu_woq.txt b/...e-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