File size: 339 Bytes
8c03069
 
 
 
 
 
 
 
 
 
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")