aman81 commited on
Commit
e8d42ae
1 Parent(s): f7a7f88

Update local_py

Browse files
Files changed (1) hide show
  1. local_app.py +3 -0
local_app.py CHANGED
@@ -11,6 +11,9 @@ from omegaconf import OmegaConf
11
  from PIL import Image
12
  from torchvision import transforms
13
  from transformers import CLIPVisionModelWithProjection
 
 
 
14
 
15
  from src.models.pose_guider import PoseGuider
16
  from src.models.unet_2d_condition import UNet2DConditionModel
 
11
  from PIL import Image
12
  from torchvision import transforms
13
  from transformers import CLIPVisionModelWithProjection
14
+ from diffusers import DiffusionPipeline
15
+
16
+ pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt")
17
 
18
  from src.models.pose_guider import PoseGuider
19
  from src.models.unet_2d_condition import UNet2DConditionModel