PRX Part 4: Our Data Strategy

Hugging Face Blog · 1 week ago

Photoroom built PRX's training dataset by mixing public and internal sources, then re-captioning all images with a vision-language model for consistency before converting everything to a standardized format. The team stored 100 million+ images in Lance format for exploration and filtering, then streamed them via Mosaic Data Shards during training, with text latents computed on-the-fly at a 3-4% throughput cost. This approach prioritized dataset breadth over individual image perfection, allowing the model to learn diverse visual concepts while keeping storage and compute efficient.