minor

a-r-r-o-w · Jan 5, 2025 · 4288f3b · 4288f3b
1 parent 1066dfc
commit 4288f3b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/training/cogvideox.md b/docs/training/cogvideox.md
@@ -118,10 +118,10 @@ Assuming your LoRA is saved and pushed to the HF Hub, and named `my-awesome-name
 
 ```diff
 import torch
-from diffusers import DiffusionPipeline
+from diffusers import CogVideoXPipeline
 from diffusers.utils import export_to_video
 
-pipe = DiffusionPipeline.from_pretrained(
+pipe = CogVideoXPipeline.from_pretrained(
     "THUDM/CogVideoX-5b", torch_dtype=torch.float16
 ).to("cuda")
 + pipe.load_lora_weights("my-awesome-name/my-awesome-lora", adapter_name="cogvideox-lora")