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
?