fix(app): import CogView4Transformer2DModel from the module that exports it

This commit is contained in:
psychedelicious
2025-04-10 06:52:08 +10:00
parent 170ea4fb75
commit ae8d1f26d6

View File

@@ -2,7 +2,7 @@ from typing import Callable, Optional
import torch
import torchvision.transforms as tv_transforms
from diffusers import CogView4Transformer2DModel
from diffusers.models.transformers.transformer_cogview4 import CogView4Transformer2DModel
from torchvision.transforms.functional import resize as tv_resize
from tqdm import tqdm