Upload ./README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -87,14 +87,14 @@ pipeline_tag: text-to-video
|
|
87 |
import torch
|
88 |
from diffusers import AutoencoderKLAllegro, AllegroPipeline
|
89 |
from diffusers.utils import export_to_video
|
90 |
-
vae = AutoencoderKLAllegro.from_pretrained("
|
91 |
vae.tile_overlap_t = 8
|
92 |
vae.tile_overlap_h = 144
|
93 |
vae.tile_overlap_w = 64
|
94 |
vae.stride = (16,112,192)
|
95 |
|
96 |
pipe = AllegroPipeline.from_pretrained(
|
97 |
-
"
|
98 |
)
|
99 |
pipe.to("cuda")
|
100 |
pipe.vae.enable_tiling()
|
|
|
87 |
import torch
|
88 |
from diffusers import AutoencoderKLAllegro, AllegroPipeline
|
89 |
from diffusers.utils import export_to_video
|
90 |
+
vae = AutoencoderKLAllegro.from_pretrained("rhymes-ai/Allegro-T2V-40x360P", subfolder="vae", torch_dtype=torch.float32)
|
91 |
vae.tile_overlap_t = 8
|
92 |
vae.tile_overlap_h = 144
|
93 |
vae.tile_overlap_w = 64
|
94 |
vae.stride = (16,112,192)
|
95 |
|
96 |
pipe = AllegroPipeline.from_pretrained(
|
97 |
+
"rhymes-ai/Allegro-T2V-40x360P", vae=vae, torch_dtype=torch.bfloat16
|
98 |
)
|
99 |
pipe.to("cuda")
|
100 |
pipe.vae.enable_tiling()
|