ValueError when using `FromOriginalModelMixin` in subclasses

[luocfprime]

Describe the bug

ValueError when using FromOriginalModelMixin in subclasses due to the following strict class check:

diffusers/src/diffusers/loaders/single_file_model.py

Lines 147 to 151 in 83bc6c9

	class_name = cls.__name__
	if class_name not in SINGLE_FILE_LOADABLE_CLASSES:
	raise ValueError(
	f"FromOriginalModelMixin is currently only compatible with {', '.join(SINGLE_FILE_LOADABLE_CLASSES.keys())}"
	)

Could it be possible to extend from_single_file to derived classes of SINGLE_FILE_LOADABLE_CLASSES too?

Reproduction

Example:

class MyVAE(AutoencoderKL):
    pass

MyVAE.from_single_file("xxx.safetensors")

Logs

No response

System Info

diffusers 0.28.0

Who can help?

@sayakpaul @DN6 @yiyixuxu

[DN6]

Hi @fkcptlst yeah I think we can make that addition. Would you like to open a PR for it? If not I can take care of it later this week.

[luocfprime]

Hi @fkcptlst yeah I think we can make that addition. Would you like to open a PR for it? If not I can take care of it later this week.

Hello, I opened a PR at #8454. Please review when you're available.

[DN6]

Hi @fkcptlst closing this since #8454 was merged to solve the issue.