from modeling_autoencoder import AutoEncoderConfig, AutoEncoder

from transformers import PretrainedConfig, PreTrainedModel, AutoConfig, AutoModel

AutoEncoderConfig.register_for_auto_class()
AutoEncoder.register_for_auto_class("AutoModel")

AutoConfig.register("autoencoder", AutoEncoderConfig)
AutoModel.register(AutoEncoderConfig, AutoModel)

AutoEncoder(AutoEncoderConfig()).push_to_hub("autoencoder")
 
# config = AutoConfig.from_pretrained("amaye15/autoencoder", trust_remote_code = True)
# autoencoder = AutoModel.from_config(config, trust_remote_code = True)