update README.md;

Signed-off-by: lawrence-cj <[email protected]>
NVlabs · Dec 13, 2024 · 548bdb8 · 548bdb8
1 parent cd5f019
commit 548bdb8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -116,7 +116,6 @@ cd Sana
 
 ## 🔛 Choose your model: [Model card](asset/docs/model_zoo.md)
 
-
 ## 🔛 Quick start with [Gradio](https://www.gradio.app/guides/quickstart)
 
 ```bash
@@ -171,7 +170,7 @@ pipe = SanaPAGPipeline.from_pretrained(
   "Efficient-Large-Model/Sana_1600M_1024px_diffusers",
   variant="fp16",
   torch_dtype=torch.float16,
-  pag_applied_layers="transformer_blocks.8"
+  pag_applied_layers="transformer_blocks.8",
 )
 pipe.to("cuda")
 

diff --git a/asset/docs/model_zoo.md b/asset/docs/model_zoo.md
@@ -56,7 +56,7 @@ pipe = SanaPAGPipeline.from_pretrained(
   "Efficient-Large-Model/Sana_1600M_1024px_BF16_diffusers",
   variant="bf16",
   torch_dtype=torch.bfloat16,
-  pag_applied_layers="transformer_blocks.8"
+  pag_applied_layers="transformer_blocks.8",
 )
 pipe.to("cuda")