| from transformers import AutoConfig | |
| # The base model you originally fine-tuned from | |
| BASE_MODEL = "Mullerjo/food-101-finetuned-model" | |
| SAVE_DIRECTORY = "./fixed-model-config" | |
| print(f"Loading the correct config from '{BASE_MODEL}'...") | |
| config = AutoConfig.from_pretrained(BASE_MODEL) | |
| print(f"Saving the correct config file to '{SAVE_DIRECTORY}'...") | |
| config.save_pretrained(SAVE_DIRECTORY) | |
| print("\nSuccess! A correct 'config.json' has been created.") |