from transformers import AutoConfig

from modeling.modeling_resnet import ResNetCustomForImageClassification

cfg = AutoConfig.from_pretrained("microsoft/resnet-18")

ResNetCustomForImageClassification.register_for_auto_class("AutoModelForImageClassification")

model = ResNetCustomForImageClassification(cfg)
model.save_pretrained("/home/fxmarty/hf_internship/tiny-testing-remote-code")