https://huggingface.co/amd/Nitro-E

우리는 고품질 이미지 생성을 위한 매우 가벼운 확산 변압기 모델인 Nitro-E를 소개합니다. 304M 매개 변수로 Nitro-E는 훈련과 추론 모두에 자원 친화적으로 설계되었습니다. 훈련의 경우 8개의 AMD Instinct™ MI300X GPU가 있는 단일 노드에서 1.5일 밖에 걸리지 않습니다. 추론 측면에서 Nitro-E는 단일 AMD Instinct MI300X GPU에서 초당 18.8 샘플(배치 크기 32, 512px 이미지)의 처리량을 제공합니다. 증류 버전은 초당 39.3 샘플로 처리량을 더 높일 수 있습니다. 소비자 iGPU 장치 Strix Halo에서 우리 모델은 단 0.16초 만에 512px 이미지를 생성할 수 있습니다. 더 자세한 비교는 그림 1에서 볼 수 있습니다. 여기서 우리의 모델은 GenEval에서 유망한 점수를 달성하면서 분명히 더 높은 처리량을 보여줍니다.

오픈 소스에 대한 AMD의 약속에 따라, 우리는 쉽게 재현할 수 있도록 모델 가중치와 전체 훈련 코드를 출시하고 있습니다. 우리는 우리의 모델이 향후 연구를 위한 강력하고 실용적인 기준이 될 수 있고 생성 AI 모델의 민주화에 더욱 기여할 수 있기를 바랍니다. 이 블로그에서는 모델 디자인 및 기타 기술적 세부 사항에 대한 일반적인 소개를 제공합니다.

AMD 지원받아 AMD 친화적으로 변형된 애들은 봤는데 이건 주체가 AMD인가 보네요.

관심 있으신 분들 구경해보세요~