From ddca99112c18dff8ea645dc700e98c1a2a4db367 Mon Sep 17 00:00:00 2001
From: Yang Yong <yongyang1030@163.com>
Date: Tue, 19 Nov 2024 23:39:16 +0800
Subject: [PATCH] fix ci (#205)

---
 ci_check/change_files.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/ci_check/change_files.py b/ci_check/change_files.py
index 2b7a0cb7..6dc4b29a 100644
--- a/ci_check/change_files.py
+++ b/ci_check/change_files.py
@@ -55,7 +55,16 @@ def main():
         elif file_path == "../llmc/compression/quantization/awq.py":
             modifications = {
                 "header": ["n_grid_zbl = 1\n"],
-                "modifications": [("n_grid = 20", "n_grid = n_grid_zbl")],
+                "modifications": [
+                    (
+                        "n_grid = 20",
+                        "n_grid = n_grid_zbl"
+                    ),
+                    (
+                        "device='cuda'",
+                        "device='cpu'"
+                    )
+                ],
             }
         elif file_path == "../llmc/compression/quantization/gptq.py":
             modifications = {