import torch
from diffusers import StableDiffusion3Pipeline # pip install diffusers>=0.31.0

pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large", torch_dtype=torch.bfloat16)
pipe.load_lora_weights("Shakker-Labs/SD3.5-LoRA-Chinese-Line-Art", weight_name="SD35-lora-Chinese-Line-Art.safetensors")
pipe.fuse_lora(lora_scale=1.0)
pipe.to("cuda")

prompt = "a boat on the river, mountain in the distance, Chinese line art"
negative_prompt = "(lowres, low quality, worst quality)"

image = pipe(prompt=prompt,
             negative_prompt=negative_prompt
             num_inference_steps=24, 
             guidance_scale=4.0,
             width=960, height=1280,
            ).images[0]
image.save(f"toy_example.jpg")