Fix an import error with PreTrainModel (#41571)

Author: remi-or

src/transformers/quantizers/base.py

@@ -32,6 +32,9 @@
 
 
 def _assign_original_dtype(module, original_dtype):
+ # not very nice in a recursive function but it avoids a circular import
+ from ..modeling_utils import PreTrainedModel
+
  for child in module.children():
  if isinstance(child, PreTrainedModel):
  child.config._pre_quantization_dtype = original_dtype