miniDiffusion: PyTorch로 구현된 최소한의 Stable Diffusion 3.5

2025-06-14
miniDiffusion: PyTorch로 구현된 최소한의 Stable Diffusion 3.5

miniDiffusion은 최소한의 종속성을 가진 순수 PyTorch를 사용하여 Stable Diffusion 3.5 모델을 간소화하여 재구현한 것입니다. 교육, 실험, 해킹 목적으로 설계되었으며, 간결한 코드베이스(약 2800줄)에는 VAE, DiT, 학습, 데이터셋 스크립트가 포함되어 있습니다. 이 프로젝트는 학습 및 추론 모두를 위한 스크립트를 제공합니다. 사용자는 종속성을 설치하고 사전 훈련된 모델의 가중치를 다운로드해야 합니다. 이 오픈소스 프로젝트는 MIT 라이선스에 따라 라이선스가 부여됩니다.

AI