Skip to content

Commit ae8d1f2

Browse files
fix(app): import CogView4Transformer2DModel from the module that exports it
1 parent 170ea4f commit ae8d1f2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

invokeai/app/invocations/cogview4_denoise.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import torch
44
import torchvision.transforms as tv_transforms
5-
from diffusers import CogView4Transformer2DModel
5+
from diffusers.models.transformers.transformer_cogview4 import CogView4Transformer2DModel
66
from torchvision.transforms.functional import resize as tv_resize
77
from tqdm import tqdm
88

0 commit comments

Comments
 (0)