Update quantization_engine.py

Author: codewithdark-git

quantllm/quant/quantization_engine.py

@@ -521,7 +521,7 @@ def _prepare_model_instance(self, model_to_prepare: PreTrainedModel, make_copy:
         prepared_model.eval()
         if self.device_manager.primary_device is not None:
             self.logger.log_info(f"Moving model to device: {self.device_manager.primary_device}")
-            prepared_model = move_to_device(prepared_model, self.device_manager.primary_device)
+            prepared_model = prepared_model.to(self.device_manager.primary_device)
 
         self.logger.log_info("Model preparation (copy, eval, device move) completed successfully.")
         return prepared_model
@@ -600,4 +600,4 @@ def _update_model_config_with_quant_params(self, method_name: str, method_specif
         else:
             self.model.config.quantization_config = quant_params
 
-        self.logger.log_info(f"Updated model.config.quantization_config with: {self.model.config.quantization_config}")
+        self.logger.log_info(f"Updated model.config.quantization_config with: {self.model.config.quantization_config}")