kuprel / min-dalle

min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Size

Muqeeth99 opened this issue · comments

What is the size of this model of min-dalle?

min-dalle uses three models in the pipeline to generate images from text.

  1. Dalle BART encoder
  2. Dalle BART decoder
  3. VQGAN Detokenizer

There are two variants of the pipeline

Variant Size Details
mini encoder - 386 MB, decoder - 448 MB, detokenizer - 178 MB
mega encoder - 2.06 GB, decoder - 2.75 GB, detokenizer - 178 MB

VQGAN Detokenizer is same in both variants.