ImportError

[seunghee-han]

Hello there!
Hello, have read your paper, and I think your paper is very good.
So I attempted to run the code, and unfortunately, I encountered an error. Here are the details:

Traceback (most recent call last):
File "./train_t2i.py", line 24, in
from dalle.models import PrefixTuningDalle, Dalle, PromptDalle, StoryDalle
ImportError: cannot import name 'PrefixTuningDalle' from 'dalle.models'

Can i get PrefixTuningDalle file?

Thank you:)