Переглянути джерело

Set low_cpu_mem_usage=True by default (fix)

Aleksandr Borzunov 2 роки тому
батько
коміт
3a37e4fa44
1 змінених файлів з 1 додано та 1 видалено
  1. 1 1
      src/bloom/model.py

+ 1 - 1
src/bloom/model.py

@@ -113,7 +113,7 @@ class _BloomPreTrainedModelWithModifiedDefaults(BloomPreTrainedModel):
     def from_pretrained(cls, *args, low_cpu_mem_usage: Optional[bool] = None, **kwargs):
         if low_cpu_mem_usage is None:
             low_cpu_mem_usage = True
-        return super().from_pretrained(*args, **kwargs)
+        return super().from_pretrained(*args, low_cpu_mem_usage=low_cpu_mem_usage, **kwargs)
 
     from_pretrained.__doc__ = BloomPreTrainedModel.from_pretrained.__doc__.replace(
         "low_cpu_mem_usage(`bool`, *optional*)",