"vscode:/vscode.git/clone" did not exist on "f7498d66f9d0ea1fb67e1bd92bf449a417cbf0e1"
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 ...@@ -2,7 +2,11 @@ import json
import os import os
import torch 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.post_infer import QwenImagePostInfer
from .infer.pre_infer import QwenImagePreInfer from .infer.pre_infer import QwenImagePreInfer
......
import json import json
import os import os
import torch # type: ignore import torch
from diffusers import AutoencoderKLQwenImage
from diffusers.image_processor import VaeImageProcessor try:
from diffusers import AutoencoderKLQwenImage
from diffusers.image_processor import VaeImageProcessor
except ImportError:
AutoencoderKLQwenImage = None
VaeImageProcessor = None
class AutoencoderKLQwenImageVAE: 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