[PR #1] [MERGED] Add a SeedVR2 checkpoint to safetensors converter #3

Closed
opened 2026-10-26 22:28:46 +00:00 by bsakaguchi · 0 comments
Owner

📋 Pull Request Information

Original PR: https://github.com/Comfy-Org/comfy-model-tools/pull/1
Author: @pollockjj
Created: 6/20/2026
Status: ✅ Merged
Merged: 6/20/2026
Merged by: @pollockjj

Base: main ← Head: seedvr2-converter-e2e9


📝 Commits (1)

  • 40cab64 Add a SeedVR2 checkpoint to safetensors converter

📊 Changes

2 files changed (+336 additions, -0 deletions)

View changed files

📝 README.md (+10 -0)
➕ seedvr2_convert.py (+326 -0)

📄 Description

A new script for packaging SeedVR2 weights for ComfyUI. There is no existing path to turn ByteDance's SeedVR2 .pth checkpoints into the safetensors ComfyUI loads; this adds one.

Containment: one new file, seedvr2_convert.py, plus a README entry describing it. No other script is touched.

It reads a SeedVR2 .pth and writes one safetensors per --job in a selected precision: fp16, fp8_e4m3fn, fp8_e4m3fn_mixed_block35_fp16, mxfp8, or nvfp4. A single source load can serve several jobs. DiT outputs embed the fixed text conditioning, so no separate conditioning file is needed at load. The mxfp8 and nvfp4 modes emit ComfyUI-native quantized weights and require the comfy-kitchen package; the script exits naming it when absent. An optional PRECISION:OUT:SHA256 verifies each written file.

Repro:

python seedvr2_convert.py --src seedvr2_ema_3b.pth --cond pos_emb.pt,neg_emb.pt \
    --job fp16:seedvr2_3b_fp16.safetensors \
    --job nvfp4:seedvr2_3b_nvfp4.safetensors

Source checkpoints: ByteDance-Seed SeedVR2 (Apache-2.0), https://huggingface.co/ByteDance-Seed/SeedVR2-3B and https://huggingface.co/ByteDance-Seed/SeedVR2-7B, pinned to the exact revisions and SHA256s recorded in the script's provenance header.


🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.

## 📋 Pull Request Information **Original PR:** https://github.com/Comfy-Org/comfy-model-tools/pull/1 **Author:** [@pollockjj](https://github.com/pollockjj) **Created:** 6/20/2026 **Status:** ✅ Merged **Merged:** 6/20/2026 **Merged by:** [@pollockjj](https://github.com/pollockjj) **Base:** `main` ← **Head:** `seedvr2-converter-e2e9` --- ### 📝 Commits (1) - [`40cab64`](https://github.com/Comfy-Org/comfy-model-tools/commit/40cab64b77f70468332b39bdaf6bb1030ee06615) Add a SeedVR2 checkpoint to safetensors converter ### 📊 Changes **2 files changed** (+336 additions, -0 deletions) <details> <summary>View changed files</summary> 📝 `README.md` (+10 -0) ➕ `seedvr2_convert.py` (+326 -0) </details> ### 📄 Description A new script for packaging SeedVR2 weights for ComfyUI. There is no existing path to turn ByteDance's SeedVR2 .pth checkpoints into the safetensors ComfyUI loads; this adds one. Containment: one new file, seedvr2_convert.py, plus a README entry describing it. No other script is touched. It reads a SeedVR2 .pth and writes one safetensors per --job in a selected precision: fp16, fp8_e4m3fn, fp8_e4m3fn_mixed_block35_fp16, mxfp8, or nvfp4. A single source load can serve several jobs. DiT outputs embed the fixed text conditioning, so no separate conditioning file is needed at load. The mxfp8 and nvfp4 modes emit ComfyUI-native quantized weights and require the comfy-kitchen package; the script exits naming it when absent. An optional PRECISION:OUT:SHA256 verifies each written file. Repro: ``` python seedvr2_convert.py --src seedvr2_ema_3b.pth --cond pos_emb.pt,neg_emb.pt \ --job fp16:seedvr2_3b_fp16.safetensors \ --job nvfp4:seedvr2_3b_nvfp4.safetensors ``` Source checkpoints: ByteDance-Seed SeedVR2 (Apache-2.0), https://huggingface.co/ByteDance-Seed/SeedVR2-3B and https://huggingface.co/ByteDance-Seed/SeedVR2-7B, pinned to the exact revisions and SHA256s recorded in the script's provenance header. --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>
bsakaguchi 2026-10-26 22:28:46 +00:00
Sign in to join this conversation.
No labels
pull-request
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
Comfy-Org/comfy-model-tools#3
No description provided.