Fix argument

7c48033a · Casper Hansen · 2dada8f8 · 7c48033a
Commit 7c48033a authored Sep 21, 2023 by Casper Hansen
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

awq/models/base.py awq/models/base.py +0 -1

No files found.
--- a/awq/models/base.py
+++ b/awq/models/base.py
@@ -122,7 +122,6 @@ class BaseAWQForCausalLM(nn.Module):
            trust_remote_code=trust_remote_code,
            torch_dtype=torch_dtype,
            use_safetensors=safetensors,
-            low_cpu_mem_usage=True,
            **model_init_kwargs
        )