Update README.md

README.md

@@ -35,7 +35,7 @@ from tuneavideo.models.unet import UNet3DConditionModel
 from tuneavideo.util import save_videos_grid
 import torch
 
-pretrained_model_path = "CompVis/stable-diffusion-v1-4"
+pretrained_model_path = "nitrosocke/mo-di-diffusion"
 unet_model_path = "Tune-A-Video-library/mo-di-bear-guitar"
 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")