refactor code

chengzeyi · Dec 29, 2024 · 687d13e · 687d13e
1 parent 7615556
commit 687d13e
Show file tree

Hide file tree

Showing 3 changed files with 778 additions and 757 deletions.
diff --git a/README.md b/README.md
@@ -299,7 +299,7 @@ vae = AutoencoderKL.from_pretrained(
 
 from para_attn.parallel_vae.diffusers_adapters import parallelize_vae
 
-vae = parallelize_vae(vae)
+parallelize_vae(vae)
 ```
 
 ## All Examples