fix ci (#205)

ModelTC · Nov 19, 2024 · ddca991 · ddca991
1 parent a6c95fc
commit ddca991
Showing 1 changed file with 10 additions and 1 deletion.
diff --git 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 = {