Load an object class from a model repo that uses `trust_remote_code=True`

I am loading a model on the hub using trust_remote_code=True.

For the sake of the example. let’s say it’s tiiuae/falcon-40b: model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-40b", trust_remote_code=True).

model is an object of class FalconForCausalLM.

What’s the simplest way i can do an from XXX import FalconForCausalLM?

Never mind, I can do a model.__class__