Update README.md (#4)
Browse files- Update README.md (b3835d5dac15cdb4a10773607e91082b30f2dee0)
Co-authored-by: Lev Novitskiy <[email protected]>
README.md
CHANGED
|
@@ -47,6 +47,9 @@ pipe = Kandinsky5T2VPipeline.from_pretrained(
|
|
| 47 |
)
|
| 48 |
pipe = pipe.to("cuda")
|
| 49 |
|
|
|
|
|
|
|
|
|
|
| 50 |
# Generate video
|
| 51 |
prompt = "A cat and a dog baking a cake together in a kitchen."
|
| 52 |
negative_prompt = "Static, 2D cartoon, cartoon, 2d animation, paintings, images, worst quality, low quality, ugly, deformed, walking backwards"
|
|
|
|
| 47 |
)
|
| 48 |
pipe = pipe.to("cuda")
|
| 49 |
|
| 50 |
+
pipe.transformer.set_attention_backend("flex")
|
| 51 |
+
pipe.transformer.compile(mode="max-autotune-no-cudagraphs", dynamic=True)
|
| 52 |
+
|
| 53 |
# Generate video
|
| 54 |
prompt = "A cat and a dog baking a cake together in a kitchen."
|
| 55 |
negative_prompt = "Static, 2D cartoon, cartoon, 2d animation, paintings, images, worst quality, low quality, ugly, deformed, walking backwards"
|