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

Commit 9b14bbe

Browse files
Merge pull request #63 from one-shot-finish/patch-1
Update source of UNet3DConditionModel
2 parents 99e5ad9 + 3f7ae63 commit 9b14bbe

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)