Update README.md
Browse files
README.md
CHANGED
@@ -36,7 +36,7 @@ Requires the [🧨 Diffusers library](https://github.com/huggingface/diffusers)
|
|
36 |
```py
|
37 |
import torch
|
38 |
from diffusers import CogVideoXPipeline
|
39 |
-
from diffusers import export_to_video
|
40 |
|
41 |
pipe = CogVideoXPipeline.from_pretrained("THUDM/CogVideoX-5b", torch_dtype=torch.bfloat16).to("cuda")
|
42 |
pipe.load_lora_weights("a-r-r-o-w/cogvideox-disney-adamw-4000-0.0003-constant", weight_name="pytorch_lora_weights.safetensors", adapter_name=["cogvideox-lora"])
|
|
|
36 |
```py
|
37 |
import torch
|
38 |
from diffusers import CogVideoXPipeline
|
39 |
+
from diffusers.utils import export_to_video
|
40 |
|
41 |
pipe = CogVideoXPipeline.from_pretrained("THUDM/CogVideoX-5b", torch_dtype=torch.bfloat16).to("cuda")
|
42 |
pipe.load_lora_weights("a-r-r-o-w/cogvideox-disney-adamw-4000-0.0003-constant", weight_name="pytorch_lora_weights.safetensors", adapter_name=["cogvideox-lora"])
|