Update all files for BitDance-14B-16x-diffusers

test_bitdance.py

@@ -0,0 +1,26 @@
+import sys
+from pathlib import Path
+
+import torch
+from diffusers import DiffusionPipeline
+
+model_path = Path(__file__).resolve().parent
+sys.path.insert(0, str(model_path))
+
+pipe = DiffusionPipeline.from_pretrained(
+    str(model_path),
+    custom_pipeline=str(model_path),
+    torch_dtype=torch.bfloat16,
+)
+pipe.to("cuda")
+
+result = pipe(
+    prompt="A cinematic landscape photo of snowy mountains at sunrise.",
+    height=512,
+    width=512,
+    num_inference_steps=30,
+    guidance_scale=7.5,
+    show_progress_bar=True,
+)
+result.images[0].save("bitdance_14b_16x.png")
+print("Saved bitdance_14b_16x.png")