Skip to content
This repository was archived by the owner on Dec 14, 2023. It is now read-only.

Commit 3f7ae63

Browse files
authored
Update source of UNet3DConditionModel
In inference, the import for UNet3DConditionModel is supposed to be from the current repo but it accidentally seem to import this class from diffusers
1 parent 99e5ad9 commit 3f7ae63

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

inference.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55
from uuid import uuid4
66
from utils.lora import inject_inferable_lora
77
import torch
8-
from diffusers import DPMSolverMultistepScheduler, TextToVideoSDPipeline, UNet3DConditionModel
8+
from diffusers import DPMSolverMultistepScheduler, TextToVideoSDPipeline
9+
from models.unet_3d_condition import UNet3DConditionModel
910
from einops import rearrange
1011
from torch.nn.functional import interpolate
1112

0 commit comments

Comments
 (0)