File size: 692 Bytes
98639dd 8c624d1 98639dd 8c624d1 98639dd 8c624d1 98639dd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from huggingface_hub import notebook_login
from transformers import AutoConfig, AutoModel
from autoencoder_model.modeling_autoencoder import AutoEncoder, AutoEncoderConfig
notebook_login()
# Register Huggingface Model
AutoEncoderConfig.register_for_auto_class()
AutoEncoder.register_for_auto_class("AutoModel")
AutoConfig.register("autoencoder", AutoEncoderConfig)
AutoModel.register(AutoEncoderConfig, AutoModel)
# Create Model
autoencoder = AutoEncoder(AutoEncoderConfig())
autoencoder.push_to_hub("autoencoder")
# Download Model
config = AutoConfig.from_pretrained("amaye15/autoencoder", trust_remote_code = True)
autoencoder = AutoModel.from_config(config, trust_remote_code = True)
|