AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

guard omnigen processor. (#11799)

Browse Source
This commit is contained in:
Sayak Paul
2025-06-24 19:15:34 +05:30
committed by GitHub
parent 5df02fc171
commit d3e27e05f0
2 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/diffusers/pipelines/omnigen/pipeline_omnigen.py
View File

@@ -23,12 +23,14 @@ from ...image_processor import PipelineImageInput, VaeImageProcessor
from ...models.autoencoders import AutoencoderKL
from ...models.transformers import OmniGenTransformer2DModel
from ...schedulers import FlowMatchEulerDiscreteScheduler
from ...utils import is_torch_xla_available, logging, replace_example_docstring
from ...utils import is_torch_xla_available, is_torchvision_available, logging, replace_example_docstring
from ...utils.torch_utils import randn_tensor
from ..pipeline_utils import DiffusionPipeline, ImagePipelineOutput
from .processor_omnigen import OmniGenMultiModalProcessor
if is_torchvision_available():
from .processor_omnigen import OmniGenMultiModalProcessor
if is_torch_xla_available():
XLA_AVAILABLE = True
else:

7
src/diffusers/pipelines/omnigen/processor_omnigen.py
View File

@@ -18,7 +18,12 @@ from typing import Dict, List
import numpy as np
import torch
from PIL import Image
from torchvision import transforms
from ...utils import is_torchvision_available
if is_torchvision_available():
from torchvision import transforms
def crop_image(pil_image, max_image_size):