Instructions to use kandinskylab/Kandinsky-WM-1.0-I2V-5s-RO with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use kandinskylab/Kandinsky-WM-1.0-I2V-5s-RO with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image, export_to_video # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("kandinskylab/Kandinsky-WM-1.0-I2V-5s-RO", dtype=torch.bfloat16, device_map="cuda") pipe.to("cuda") prompt = "A man with short gray hair plays a red electric guitar." image = load_image( "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png" ) output = pipe(image=image, prompt=prompt).frames[0] export_to_video(output, "output.mp4") - Notebooks
- Google Colab
- Kaggle
Commit 路
7989d31
1
Parent(s): 0fdc7ef
Update project links and hardware note
Browse files
README.md
CHANGED
|
@@ -14,7 +14,9 @@ tags: [kandinsky, image-to-video, physical-ai, world-model]
|
|
| 14 |
**Image-to-Video generation for Physical AI: autonomous driving 路 robotics 路 general physics**
|
| 15 |
|
| 16 |
[馃 Checkpoints collection](https://huggingface.co/collections/kandinskylab/kandinsky-wm-10-6a67455245052645fcfbd5f3) 路
|
| 17 |
-
[馃З Base model (Kandinsky 5.0)](https://github.com/kandinskylab/kandinsky-5)
|
|
|
|
|
|
|
| 18 |
|
| 19 |
</div>
|
| 20 |
|
|
@@ -220,8 +222,7 @@ frames = pipe(
|
|
| 220 |
export_to_video(frames, "robotics_out.mp4", fps=24, quality=9)
|
| 221 |
```
|
| 222 |
|
| 223 |
-
Script: [`examples/run_i2v_diffusers.py`](https://github.com/kandinskylab/kandinsky-wm/blob/main/examples/run_i2v_diffusers.py).
|
| 224 |
-
fits a single 80 GB GPU with no offload; ~4 min for 50 steps on one H100.
|
| 225 |
|
| 226 |
|
| 227 |
## Acknowledgements
|
|
|
|
| 14 |
**Image-to-Video generation for Physical AI: autonomous driving 路 robotics 路 general physics**
|
| 15 |
|
| 16 |
[馃 Checkpoints collection](https://huggingface.co/collections/kandinskylab/kandinsky-wm-10-6a67455245052645fcfbd5f3) 路
|
| 17 |
+
[馃З Base model (Kandinsky 5.0)](https://github.com/kandinskylab/kandinsky-5) 路
|
| 18 |
+
[馃捇 GitHub](https://github.com/kandinskylab/kandinsky-wm) 路
|
| 19 |
+
[馃摑 Habr article](https://habr.com/ru/companies/sberbank/articles/1054966/)
|
| 20 |
|
| 21 |
</div>
|
| 22 |
|
|
|
|
| 222 |
export_to_video(frames, "robotics_out.mp4", fps=24, quality=9)
|
| 223 |
```
|
| 224 |
|
| 225 |
+
Script: [`examples/run_i2v_diffusers.py`](https://github.com/kandinskylab/kandinsky-wm/blob/main/examples/run_i2v_diffusers.py).
|
|
|
|
| 226 |
|
| 227 |
|
| 228 |
## Acknowledgements
|