Update README.md

nateraw · Oct 20, 2022 · bdc7917 · bdc7917
1 parent 9a3919e
commit bdc7917
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -115,8 +115,16 @@ video_path = pipeline.walk(
 #### Run the App Locally
 
 ```python
-from stable_diffusion_videos import interface
+from stable_diffusion_videos import StableDiffusionWalkPipeline
+import torch
+
+pipeline = StableDiffusionWalkPipeline.from_pretrained(
+    "CompVis/stable-diffusion-v1-4",
+    torch_dtype=torch.float16,
+    revision="fp16",
+).to("cuda")
 
+interface = Interface(pipeline)
 interface.launch()
 ```