BitDance-14B-16x-diffusers / load_pipeline.py
BiliSakura's picture
Update all files for BitDance-14B-16x-diffusers
8c03069 verified
Raw
History Blame
339 Bytes
import sys
from pathlib import Path
from diffusers import DiffusionPipeline
model_dir = Path(__file__).resolve().parent
sys.path.insert(0, str(model_dir))
pipe = DiffusionPipeline.from_pretrained(model_dir, trust_remote_code=True).to("cuda")
images = pipe(prompt="A scenic mountain lake at sunrise.").images
images[0].save("sample.png")