Skip to content

Commit 4a06c1c

Browse files
committed
test
1 parent 09359e6 commit 4a06c1c

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

src/diffusers/pipelines/pipeline_utils.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -867,7 +867,16 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
867867
expected_types = pipeline_class._get_signature_types()
868868
passed_class_obj = {k: kwargs.pop(k) for k in expected_modules if k in kwargs}
869869
passed_pipe_kwargs = {k: kwargs.pop(k) for k in optional_kwargs if k in kwargs}
870-
init_dict, unused_kwargs, _ = pipeline_class.extract_init_dict(config_dict, **kwargs)
870+
#init_dict, unused_kwargs, _ = pipeline_class.extract_init_dict(config_dict, **kwargs, **passed_pipe_kwargs)
871+
init_dict = {}
872+
unused_kwargs = {}
873+
_ = expected_types.pop("self")
874+
for key, model_class in expected_types.items():
875+
class_str = str(model_class[0])
876+
class_split = class_str.split(".")
877+
library_name = class_split[0].split("'")[-1]
878+
class_name = class_split[-1].split("'")[0]
879+
init_dict[key] = (library_name, class_name,)
871880

872881
# define init kwargs and make sure that optional component modules are filtered out
873882
init_kwargs = {

0 commit comments

Comments
 (0)