LarryTsai commited on
Commit
0023259
·
verified ·
1 Parent(s): 6f08040

Upload ./README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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("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
- "allegro_t2v_40x360p", vae=vae, torch_dtype=torch.bfloat16
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()