mingyu-oo/stable-diffusion-v1-5-lora-hyundai-car-design

LoRA 기반 Stable Diffusion v1.5 자동차 이미지 파인튜닝 모델 🚗

이 모델은 sd-legacy/stable-diffusion-v1-5를 기반으로,
현대자동차 이미지와 해당 디자인을 설명한 텍스트로 LoRA 방식의 파인튜닝을 진행한 모델입니다.

총 58개의 이미지-텍스트 쌍을 사용해 학습을 진행했으며, train_text_to_image_lora.py 스크립트를 활용하여 아래와 같은 설정으로 학습했습니다.

---

🔧 학습 설정

• 해상도: 512 x 512
• 배치 크기: 1 (gradient accumulation steps = 4)
• 총 학습 스텝 수: 1000 steps
• 체크포인트 저장 주기: 100 step마다 저장
• 혼합 정밀도: fp16
• 시드 값: 42

---

💡 사용 방법

from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained(
    "sd-legacy/stable-diffusion-v1-5",
    torch_dtype=torch.float16
).to("cuda")

pipe.unet.load_attn_procs("your_username_or_org/lora-hyundai-sd15/checkpoint-XXX")

prompt = "Front of SUV"
image = pipe(prompt).images[0]
image.save("output.png")