test

war3et · Aug 7, 2024 · 8896057 · 8896057
1 parent bfe8c41
commit 8896057
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ppdiffusers/examples/inference/class_conditional_image_generation-dit.py b/ppdiffusers/examples/inference/class_conditional_image_generation-dit.py
@@ -13,8 +13,10 @@
 # limitations under the License.
 
 import os
+
 import paddle
 from paddlenlp.trainer import set_seed
+
 from ppdiffusers import DDIMScheduler, DiTPipeline
 
 os.environ["INFOPTIMIZE"] = "False"
@@ -27,6 +29,5 @@
 words = ["golden retriever"]  # class_ids [207]
 class_ids = pipe.get_label_ids(words)
 
-
 image = pipe(class_labels=class_ids, num_inference_steps=25).images[0]
-image.save("class_conditional_image_generation-dit-result.png")
+image.save("class_conditional_image_generation-dit-result.png")