HF RL Explorer

Feature Request: AutoModel class for agnostic loading of diffusers models

Feature Request: AutoModel class for agnostic loading of diffusers models: a task in LegoFlow-SWE (Harbor dataset). Currently, users who want to load individual model components (e.g., transformers, VAEs, UNets) to pass into pipelines must know the exact model class:

The task

Currently, users who want to load individual model components (e.g., transformers, VAEs, UNets) to pass into pipelines must know the exact model class:

Part of Lego-X/LegoFlow-SWE.