from kosmos2_5.tasks import GenerationTask
from kosmos2_5.models import UniGPTmodel