Load an object class from a model repo that uses `trust_remote_code=True`
I am loading a model on the hub usingtrust_remote_code=True.For the sake of the example. let’s say it’stiiuae/falcon-40b:model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-40b", trust_remote_code=True).modelis an object of classFalconForCausalLM.What’s the simplest way i can do anfrom XXX import FalconForCausalLM?
2023-11-22T13:44:54Z
Never mind, I can do amodel.__class__