Skip to content

Commit 337b2fc

Browse files
committed
fix default value
1 parent 21f94a1 commit 337b2fc

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/diffusers/models/modeling_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -690,7 +690,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
690690
device_map = kwargs.pop("device_map", None)
691691
max_memory = kwargs.pop("max_memory", None)
692692
offload_folder = kwargs.pop("offload_folder", None)
693-
offload_state_dict = kwargs.pop("offload_state_dict", False)
693+
offload_state_dict = kwargs.pop("offload_state_dict", None)
694694
low_cpu_mem_usage = kwargs.pop("low_cpu_mem_usage", _LOW_CPU_MEM_USAGE_DEFAULT)
695695
variant = kwargs.pop("variant", None)
696696
use_safetensors = kwargs.pop("use_safetensors", None)

src/diffusers/pipelines/pipeline_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -679,7 +679,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
679679
device_map = kwargs.pop("device_map", None)
680680
max_memory = kwargs.pop("max_memory", None)
681681
offload_folder = kwargs.pop("offload_folder", None)
682-
offload_state_dict = kwargs.pop("offload_state_dict", False)
682+
offload_state_dict = kwargs.pop("offload_state_dict", None)
683683
low_cpu_mem_usage = kwargs.pop("low_cpu_mem_usage", _LOW_CPU_MEM_USAGE_DEFAULT)
684684
variant = kwargs.pop("variant", None)
685685
dduf_file = kwargs.pop("dduf_file", None)

0 commit comments

Comments
 (0)