Update all files for BitDance-14B-16x-diffusers
Browse files
README.md
CHANGED
|
@@ -23,11 +23,11 @@ Diffusers-converted checkpoint for BitDance-14B-16x with bundled custom pipeline
|
|
| 23 |
import torch
|
| 24 |
from diffusers import DiffusionPipeline
|
| 25 |
|
| 26 |
-
|
| 27 |
-
|
| 28 |
pipe = DiffusionPipeline.from_pretrained(
|
| 29 |
-
|
| 30 |
-
|
| 31 |
torch_dtype=torch.bfloat16,
|
| 32 |
).to("cuda")
|
| 33 |
|
|
@@ -41,6 +41,32 @@ result = pipe(
|
|
| 41 |
result.images[0].save("bitdance_14b_16x.png")
|
| 42 |
```
|
| 43 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
## Model Metadata
|
| 45 |
|
| 46 |
- Pipeline class: `BitDanceDiffusionPipeline`
|
|
|
|
| 23 |
import torch
|
| 24 |
from diffusers import DiffusionPipeline
|
| 25 |
|
| 26 |
+
# Local path (recommended - no trust_remote_code needed)
|
| 27 |
+
model_path = "path/to/BitDance-14B-16x-diffusers"
|
| 28 |
pipe = DiffusionPipeline.from_pretrained(
|
| 29 |
+
model_path,
|
| 30 |
+
custom_pipeline=model_path,
|
| 31 |
torch_dtype=torch.bfloat16,
|
| 32 |
).to("cuda")
|
| 33 |
|
|
|
|
| 41 |
result.images[0].save("bitdance_14b_16x.png")
|
| 42 |
```
|
| 43 |
|
| 44 |
+
## Test Running
|
| 45 |
+
|
| 46 |
+
Run tests from the model directory in your active Python environment:
|
| 47 |
+
|
| 48 |
+
```bash
|
| 49 |
+
python test_bitdance.py
|
| 50 |
+
```
|
| 51 |
+
|
| 52 |
+
## VRAM Usage by Resolution
|
| 53 |
+
|
| 54 |
+
Measured on `NVIDIA A100-SXM4-80GB` using:
|
| 55 |
+
|
| 56 |
+
- `dtype=torch.bfloat16`
|
| 57 |
+
- `num_inference_steps=30`
|
| 58 |
+
- `guidance_scale=7.5`
|
| 59 |
+
- prompt: `A cinematic landscape photo of snowy mountains at sunrise.`
|
| 60 |
+
|
| 61 |
+
| Resolution | Peak Allocated VRAM (GiB) | Peak Reserved VRAM (GiB) | Time (s) | Status |
|
| 62 |
+
| --- | ---: | ---: | ---: | --- |
|
| 63 |
+
| 512x512 | 32.67 | 33.47 | 13.71 | ok |
|
| 64 |
+
| 1024x1024 | 35.51 | 38.76 | 54.47 | ok |
|
| 65 |
+
| 1280x768 | 35.28 | 38.34 | 50.97 | ok |
|
| 66 |
+
| 768x1280 | 35.28 | 38.34 | 51.22 | ok |
|
| 67 |
+
| 1536x640 | 35.28 | 38.34 | 51.29 | ok |
|
| 68 |
+
| 2048x512 | 35.51 | 38.76 | 54.61 | ok |
|
| 69 |
+
|
| 70 |
## Model Metadata
|
| 71 |
|
| 72 |
- Pipeline class: `BitDanceDiffusionPipeline`
|