Instructions to use BiliSakura/BitDance-14B-64x-diffusers with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use BiliSakura/BitDance-14B-64x-diffusers with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("BiliSakura/BitDance-14B-64x-diffusers", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps
- Draw Things
- DiffusionBee
Update all files for BitDance-14B-64x-diffusers
Browse files
bitdance_diffusers/convert.py
CHANGED
|
@@ -48,7 +48,7 @@ def _copy_runtime_source(output_path: Path) -> None:
|
|
| 48 |
"",
|
| 49 |
"model_dir = Path(__file__).resolve().parent",
|
| 50 |
"sys.path.insert(0, str(model_dir))",
|
| 51 |
-
'pipe = DiffusionPipeline.from_pretrained(model_dir,
|
| 52 |
'images = pipe(prompt="A scenic mountain lake at sunrise.").images',
|
| 53 |
'images[0].save("sample.png")',
|
| 54 |
]
|
|
|
|
| 48 |
"",
|
| 49 |
"model_dir = Path(__file__).resolve().parent",
|
| 50 |
"sys.path.insert(0, str(model_dir))",
|
| 51 |
+
'pipe = DiffusionPipeline.from_pretrained(model_dir, custom_pipeline=model_dir).to("cuda")',
|
| 52 |
'images = pipe(prompt="A scenic mountain lake at sunrise.").images',
|
| 53 |
'images[0].save("sample.png")',
|
| 54 |
]
|