from .tasks import GenerationTask
from .models import UniGPTmodel