add more requirements

Signed-off-by: yiliu30 <[email protected]>
intel · Dec 28, 2024 · 7703d97 · 7703d97
1 parent 034efb4
commit 7703d97
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/3x/torch/requirements.txt b/test/3x/torch/requirements.txt
@@ -1,4 +1,6 @@
+accelerate
 auto_round_lib @ git+https://github.com/intel/auto-round.git@xuehao/fix_install
+datasets
 deepspeed @ git+https://github.com/HabanaAI/[email protected]
 expecttest
 intel_extension_for_pytorch