[community integration]native diffusers implementation on huggingface model #3

import torch
from diffusers import DiffusionPipeline

pipe = DiffusionPipeline.from_pretrained(
    "BiliSakura/JLT-diffusers",
    subfolder="JLT-B-1",  # or "JLT-L-1"
    custom_pipeline="pipeline.py",
    trust_remote_code=True,
    torch_dtype=torch.bfloat16,
).to("cuda")

print(pipe.id2label[207])
image = pipe(
    class_labels="golden retriever",
    num_inference_steps=50,
    guidance_scale=2.9,
    generator=torch.Generator(device="cuda").manual_seed(42),
).images[0]
image.save("demo.png")