DallE mega use in transformers

🤗Transformers
1

Hi,
I was trying to use DallE mega with transformers. I was able to use other models of bloom etc with transformer, but the DallE Mega documentation mentioned here dalle-mini/dalle-mega · Hugging Face with code snippet as

from transformers import eBart

model = eBart.from_pretrained("dalle-mini/dalle-mega")

has been failing with error as below

----> 1 from transformers import eBart
      3 # Sentences we want sentence embeddings for
      4 sentences = ['A king with apple on head.']

**ImportError: cannot import name 'eBart' from 'transformers'**

Can anyone help who could use DallE Mega?

1 Like
2

Hi,

We’re currently adding support for it: dalle mega by patil-suraj · Pull Request #18152 · huggingface/transformers · GitHub.

cc @valhalla