automatic/modules/omnigen2/__init__.py

4 lines
168 B
Python

from transformers import Qwen2_5_VLForConditionalGeneration
from .pipeline_omnigen2 import OmniGen2Pipeline
from .models.transformers import OmniGen2Transformer2DModel