Use accelerate for spacy-llm to offload to CPU via device_map

[sus-domesticus]

I want to integrate an LLM into my spacy pipeline but my GPU doesn't have enough space.

I read here that accelerate could be used with the option device_map=auto to offload to the CPU but I don't know how to adapt a basic example to do that.

For example how would you do this for the following example (taken from here):

from spacy_llm.util import assemble

nlp = assemble("config.cfg")
doc = nlp("Jack and Jill rode up the hill in Les Deux Alpes")
print([(ent.text, ent.label_) for ent in doc.ents])