fix failing tests

huggingface · Aug 22, 2024 · 6d7bf69 · 6d7bf69
1 parent c7e0ec0
commit 6d7bf69
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/.github/workflows/update_llm_perf_intel_pytorch.yml b/.github/workflows/update_llm_perf_intel_pytorch.yml
@@ -17,7 +17,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        subset: [unquantized, bnb, awq, gptq]
+        subset: [unquantized]
 
         machine: [
           {name: c7i, runs-on: {group: 'aws-c7i-8xlarge-plus'}}, 
@@ -48,7 +48,7 @@ jobs:
             --volume ${{ github.workspace }}:/workspace
             --workdir /workspace
           run: |
-            pip install packaging && pip install flash-attn einops scipy auto-gptq optimum bitsandbytes autoawq codecarbon
+            pip install packaging && pip install einops scipy optimum codecarbon
             pip install -U transformers huggingface_hub[hf_transfer]
             pip install -e .
             python llm_perf/hardware/intel/update_llm_perf_intel_pytorch.py
diff --git a/llm_perf/hardware/intel/update_llm_perf_intel_pytorch.py b/llm_perf/hardware/intel/update_llm_perf_intel_pytorch.py
@@ -127,7 +127,6 @@ def benchmark_intel_pytorch(model, attn_implementation, weights_config):
     backend_config = PyTorchConfig(
         model=model,
         device="cpu",
-        device_ids="0",
         no_weights=True,
         library="transformers",
         task="text-generation",