Commit 3d8cb02e authored by helloyongyang's avatar helloyongyang
Browse files

fix import errors from qwen-image model

parent c7bfdb1c
......@@ -2,7 +2,11 @@ import json
import os
import torch
from diffusers.models.transformers.transformer_qwenimage import QwenImageTransformer2DModel
try:
from diffusers.models.transformers.transformer_qwenimage import QwenImageTransformer2DModel
except ImportError:
QwenImageTransformer2DModel = None
from .infer.post_infer import QwenImagePostInfer
from .infer.pre_infer import QwenImagePreInfer
......
import json
import os
import torch # type: ignore
from diffusers import AutoencoderKLQwenImage
from diffusers.image_processor import VaeImageProcessor
import torch
try:
from diffusers import AutoencoderKLQwenImage
from diffusers.image_processor import VaeImageProcessor
except ImportError:
AutoencoderKLQwenImage = None
VaeImageProcessor = None
class AutoencoderKLQwenImageVAE:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment