-
Notifications
You must be signed in to change notification settings - Fork 6.1k
Closed
Description
current AutoModel implementation only support models that importable from diffusers
diffusers/src/diffusers/models/auto_model.py
Line 164 in 448c72a
model_cls = getattr(library, orig_class_name, None) |
it uses this field in model config.json https://huggingface.co/HiDream-ai/HiDream-I1-Full/blob/main/transformer/config.json#L2
we can use info in model_index.json instead, https://huggingface.co/HiDream-ai/HiDream-I1-Full/blob/main/model_index.json
this way we can load anything that can be load in from_pretrained
, include transformer models & custom diffusers moduels (not directly importable from top level)
can reference the code/logic in from_pretrained
init_dict = {k: v for k, v in init_dict.items() if load_module(k, v)} |
I will work on this shortly, but feel free to pick this up if anyone in the community is interested!
sayakpaul
Metadata
Metadata
Assignees
Type
Projects
Status
Done