Remove all traces of low_cpu_mem_usage (#38792)
* remove it from all py files * remove it from the doc * remove it from examples * style * remove traces of _fast_init * Update test_peft_integration.py * CIs
This commit is contained in:
@@ -328,7 +328,6 @@ class AwqFusedTest(unittest.TestCase):
|
||||
model = AutoModelForCausalLM.from_pretrained(
|
||||
self.model_name,
|
||||
quantization_config=quantization_config,
|
||||
low_cpu_mem_usage=True,
|
||||
revision=self.model_revision,
|
||||
).to(torch_device)
|
||||
|
||||
@@ -347,7 +346,6 @@ class AwqFusedTest(unittest.TestCase):
|
||||
model = AutoModelForCausalLM.from_pretrained(
|
||||
model_id,
|
||||
quantization_config=quantization_config,
|
||||
low_cpu_mem_usage=True,
|
||||
).to(torch_device)
|
||||
|
||||
# Check if model has been correctly fused
|
||||
@@ -370,7 +368,6 @@ class AwqFusedTest(unittest.TestCase):
|
||||
model = AutoModelForCausalLM.from_pretrained(
|
||||
self.model_name,
|
||||
quantization_config=quantization_config,
|
||||
low_cpu_mem_usage=True,
|
||||
revision=self.model_revision,
|
||||
).to(torch_device)
|
||||
|
||||
@@ -399,7 +396,6 @@ class AwqFusedTest(unittest.TestCase):
|
||||
model = AutoModelForCausalLM.from_pretrained(
|
||||
self.model_name,
|
||||
quantization_config=quantization_config,
|
||||
low_cpu_mem_usage=True,
|
||||
revision=self.model_revision,
|
||||
).to(torch_device)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user