diff --git a/optimum/neuron/modeling_diffusion.py b/optimum/neuron/modeling_diffusion.py index d8b7fbfca..86179aab5 100644 --- a/optimum/neuron/modeling_diffusion.py +++ b/optimum/neuron/modeling_diffusion.py @@ -235,7 +235,7 @@ def _from_pretrained( vae_decoder_file_name: Optional[str] = NEURON_FILE_NAME, local_files_only: bool = False, model_save_dir: Optional[Union[str, Path, TemporaryDirectory]] = None, - device_ids: Optional[List[int]] = None, + device_ids: Optional[List[int]] = [], **kwargs, # To share kwargs only available for `_from_transformers` ): model_id = str(model_id) @@ -361,7 +361,7 @@ def _from_transformers( disable_fast_relayout: Optional[bool] = False, disable_fallback: bool = False, dynamic_batch_size: bool = False, - device_ids: Optional[List[int]] = None, + device_ids: Optional[List[int]] = [], **kwargs_shapes, ) -> "NeuronStableDiffusionPipelineBase": if task is None: