File size: 339 Bytes
340da85 | 1 2 3 4 5 6 7 8 9 10 11 | 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")
|