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")