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