from transformers import AutoConfig

from modeling import ResNetCustomForImageClassification

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

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