Delete stale files for BitDance-14B-16x-diffusers
Browse files- load_pipeline.py +0 -10
load_pipeline.py
DELETED
|
@@ -1,10 +0,0 @@
|
|
| 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")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|