Calling the model after registering it with automodel.register
I have registered my model using 'AutoModelForCausalLM.register(CustomAIConfig, CustomAI) ’ . But at the time of calling the model from huggingface hub it shows ’ ImportError: cannot import name ‘CustomAI’ from ‘transformers’ (/usr/local/lib/python3.10/dist-packages/transformers/init.py) '. Can anyone guide me how can I do this?