kyujinpy
/

Tune-A-VideKO-v1-5

TuneAVideoPipeline

Model card Files Files and versions Community

kyujinpy commited on Aug 14, 2023

Commit

f7c7d73

·

1 Parent(s): 050686b

Upload README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ from tuneavideo.util import save_videos_grid
 import torch
 pretrained_model_path = "Bingsu/my-korean-stable-diffusion-v1-5"
-unet_model_path = "kyujinpy/Tune-A-VideoKO-v1-5"
 unet = UNet3DConditionModel.from_pretrained(unet_model_path, subfolder='unet', torch_dtype=torch.float16).to('cuda')
 pipe = TuneAVideoPipeline.from_pretrained(pretrained_model_path, unet=unet, torch_dtype=torch.float16).to("cuda")
 pipe.enable_xformers_memory_efficient_attention()

 import torch
 pretrained_model_path = "Bingsu/my-korean-stable-diffusion-v1-5"
+unet_model_path = "kyujinpy/Tune-A-VideKO-v1-5"
 unet = UNet3DConditionModel.from_pretrained(unet_model_path, subfolder='unet', torch_dtype=torch.float16).to('cuda')
 pipe = TuneAVideoPipeline.from_pretrained(pretrained_model_path, unet=unet, torch_dtype=torch.float16).to("cuda")
 pipe.enable_xformers_memory_efficient_attention()