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

README.md

@@ -23,16 +23,16 @@ Diffusers-converted checkpoint for BitDance-14B-64x with bundled custom pipeline
 import torch
 from diffusers import DiffusionPipeline
 
-repo_id = "BiliSakura/BitDance-14B-64x-diffusers"
-
+# Local path (recommended - no trust_remote_code needed)
+model_path = "path/to/BitDance-14B-64x-diffusers"
 pipe = DiffusionPipeline.from_pretrained(
-    repo_id,
-    trust_remote_code=True,
+    model_path,
+    custom_pipeline=model_path,
     torch_dtype=torch.bfloat16,
 ).to("cuda")
 
 result = pipe(
-    prompt="A cinematic landscape photo of snowy mountains at sunrise.",
+    prompt = "A close-up portrait in a cinematic photography style, capturing a girl-next-door look on a sunny daytime urban street. She wears a khaki sweater, with long, flowing hair gently draped over her shoulders. Her head is turned slightly, revealing soft facial features illuminated by realistic, delicate sunlight coming from the left. The sunlight subtly highlights individual strands of her hair. The image has a Canon film-like color tone, evoking a warm nostalgic atmosphere.",
     height=1024,
     width=1024,
     num_inference_steps=50,
@@ -41,6 +41,32 @@ result = pipe(
 result.images[0].save("bitdance_14b_64x.png")
 ```
 
+## Test Running
+
+Run tests from the model directory in your active Python environment:
+
+```bash
+python test_bitdance.py
+```
+
+## VRAM Usage by Resolution
+
+Measured on `NVIDIA A100-SXM4-80GB` using:
+
+- `dtype=torch.bfloat16`
+- `num_inference_steps=30`
+- `guidance_scale=7.5`
+- prompt: `A close-up portrait in a cinematic photography style, capturing a girl-next-door look on a sunny daytime urban street. She wears a khaki sweater, with long, flowing hair gently draped over her shoulders. Her head is turned slightly, revealing soft facial features illuminated by realistic, delicate sunlight coming from the left. The sunlight subtly highlights individual strands of her hair. The image has a Canon film-like color tone, evoking a warm nostalgic atmosphere.`
+
+| Resolution | Peak Allocated VRAM (GiB) | Peak Reserved VRAM (GiB) | Time (s) | Status |
+| --- | ---: | ---: | ---: | --- |
+| 512x512 | 39.60 | 40.62 | 4.08 | ok |
+| 1024x1024 | 41.21 | 50.15 | 15.79 | ok |
+| 1280x768 | 40.88 | 49.52 | 14.78 | ok |
+| 768x1280 | 40.88 | 49.52 | 14.75 | ok |
+| 1536x640 | 40.88 | 49.52 | 14.76 | ok |
+| 2048x512 | 41.21 | 50.15 | 15.85 | ok |
+
 ## Model Metadata
 
 - Pipeline class: `BitDanceDiffusionPipeline`