Model reference · open weights
OpenSora-STDiT-stage2 is an open-weight embedding model from hpcai-tech. OpenSora-STDiT-v2-stage2 (FP32) weighs 1.5 GB; the smallest configuration that runs it is RTX 3060 12 GB.
What it is
| Released by | hpcai-tech |
|---|---|
| Type | Embedding models |
| Task | Embeddings |
| Parameters (lead) | 768M |
| Runs with | transformers |
| Released | 2024-04-24 |
| Popularity | 9 downloads / month |
| Weights | 1.5 GB (OpenSora-STDiT-v2-stage2 (FP32), file size) |
| Licence | Open weights |
What it runs on
Weights 1.5 GB (file size) · overhead about 1.1 GB.
| Card | Runs | Counted memory |
|---|---|---|
| RTX 3060 12 GB | fits | 11.6 GB |
| RTX 4060 Ti 16 GB | fits | 15.4 GB |
| RTX 3090 24 GB | fits | 23.4 GB |
| RTX 4090 24 GB | fits | 23.4 GB |
| RTX 5090 32 GB | fits | 31.0 GB |
| L40S 48 GB | fits | 44.0 GB |
| A100 80 GB | fits | 78.2 GB |
| H100 80 GB | fits | 78.1 GB |
| RTX PRO 6000 Blackwell 96 GB | fits | 93.8 GB |
| DGX Spark (GB10) 128 GB unified | fits | 107 GB |
| H200 141 GB | fits | 138 GB |
| B200 180 GB | fits | 176 GB |
Estimates, not measurements: the weights are the build's file size. No cache grows with use; a batch of inputs needs working memory of its own. Counted memory is 92 % of what CUDA reports for the card.