Skip to content

Commit 5eb5aaf

Browse files
committed
fix
1 parent 8ec3f26 commit 5eb5aaf

2 files changed

Lines changed: 3 additions & 0 deletions

File tree

src/diffusers/models/auto_model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -320,6 +320,7 @@ def from_pretrained(cls, pretrained_model_or_path: str | os.PathLike | None = No
320320
subfolder=subfolder,
321321
module_file=module_file,
322322
class_name=class_name,
323+
trust_remote_code=trust_remote_code,
323324
**hub_kwargs,
324325
)
325326
else:

tests/models/test_models_auto.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,7 @@ def test_from_config_with_dict_diffusers_class(self, mock_get_class):
9999
importable_classes=unittest.mock.ANY,
100100
pipelines=None,
101101
is_pipeline_module=False,
102+
trust_remote_code=False,
102103
)
103104
mock_get_class.return_value[0].from_config.assert_called_once_with(config)
104105
assert result is mock_model
@@ -139,6 +140,7 @@ def test_from_config_with_model_type_routes_to_transformers(self, mock_get_class
139140
importable_classes=unittest.mock.ANY,
140141
pipelines=None,
141142
is_pipeline_module=False,
143+
trust_remote_code=False,
142144
)
143145
assert result is mock_model
144146

0 commit comments

Comments
 (0)