소스 검색

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*)",