BiliSakura commited on
Commit
340da85
·
verified ·
1 Parent(s): 1e107ad

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

Browse files
Files changed (1) hide show
  1. load_pipeline.py +10 -0
load_pipeline.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ import sys
2
+ from pathlib import Path
3
+
4
+ from diffusers import DiffusionPipeline
5
+
6
+ model_dir = Path(__file__).resolve().parent
7
+ sys.path.insert(0, str(model_dir))
8
+ pipe = DiffusionPipeline.from_pretrained(model_dir, trust_remote_code=True).to("cuda")
9
+ images = pipe(prompt="A scenic mountain lake at sunrise.").images
10
+ images[0].save("sample.png")