BiliSakura commited on
Commit
30543b5
·
verified ·
1 Parent(s): 8105db9

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

Browse files
Files changed (1) hide show
  1. README.md +30 -4
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
- repo_id = "BiliSakura/BitDance-14B-16x-diffusers"
27
-
28
  pipe = DiffusionPipeline.from_pretrained(
29
- repo_id,
30
- trust_remote_code=True,
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`